How use properly tensorflow functions within the model

Question

I\'m using the functional API of TensorFlow 2 and tensorflow.keras.layers to build the model.

I have an input tensor (in_1) with shape [batch_size, length, di