-1
我有2個形狀的張量(10, 100, 2)
我如何得到形狀的均方損失(10, 100)
?如何從兩張張量中獲得均方?
這是我曾嘗試:
tf.sqrt(tf.square(tf.subtract(targets, logits))/n)
返回(10, 100, 2)
我有2個形狀的張量(10, 100, 2)
我如何得到形狀的均方損失(10, 100)
?如何從兩張張量中獲得均方?
這是我曾嘗試:
tf.sqrt(tf.square(tf.subtract(targets, logits))/n)
返回(10, 100, 2)
只需使用axis
參數tf.reduce_mean
:
x = tf.reduce_mean(tf.sqrt(tf.subtract(targets, logits)), axis=2)
x.shape =(10,100)
爲什麼我我低調了?我相信很多人都有同樣的問題。 –
'loss = tf.map_fn(lambda x:tf.map_fn(lambda y:tf.nn.l2_loss(y),x),tf.subtract(targets,logits))' –