how to use Lambda function correctly with custom function in tensorflow?

Question

I tried to call the custom function in tf.keras.layers.Lambda function, but I have zero gradient issues when I trained the model, don\'t know why I have this issue.