В неигровом режиме:
def gradients:
U_pred=tf.gradients(u,x)
return U_pred #returns 100x200 list (expected behaviour)
В режиме ожидания TF2:
def gradients:
with tf.GradientTape() as g:
g.watch(x)
U_pred=g.gradient(u,x)
return U_pred #return 100X1 list (expected 100x200)
Как получить список 100x200 в режиме ожидания? Есть ли способ?