How to use tf.py_func in keras lambda layer to wrap python code. ValueError: The last dimension of the inputs to Dense should be defined. Found None

Question

I would like to wrap numpy codes using tf.py_fucn in a customized lambda layer using keras.

Notice: only for simplicity I\'m showing a simple np.power function.

<