ValueError raised when tried to use custom function on tensorflow Lambda layer

Question

I tried to use the custom function on tensorflow Lambda layer but got a value error which indicates that there is a dimension error. For my custom function, RGB ten