I\'m trying to create a custom layer for my model, which can be used the classic Dense layer of Keras. Here my custom layer:
class MyDenseLayer(tf.keras.layers.La