Build a matrix from keras weights in call method

I have the next function:

Reinforcement Learning Shape Error: ValueError: Error

I am using the below approach to create simple DL-RL Model, but I am getting this error: