How to convert this numpy to tf.function compatible code?

Question

I\'m converting numpy to tensorflow equivalent code to be compatible with tf.function ...

Given have a (32, 6) numpy array target_values tha