2017-06-17 65 views
-1

我有2個形狀的張量(10, 100, 2) 我如何得到形狀的均方損失(10, 100)如何從兩張張量中獲得均方?

這是我曾嘗試:

tf.sqrt(tf.square(tf.subtract(targets, logits))/n)返回(10, 100, 2)

+0

爲什麼我我低調了?我相信很多人都有同樣的問題。 –

+0

'loss = tf.map_fn(lambda x:tf.map_fn(lambda y:tf.nn.l2_loss(y),x),tf.subtract(targets,logits))' –

回答

0

只需使用axis參數tf.reduce_mean

x = tf.reduce_mean(tf.sqrt(tf.subtract(targets, logits)), axis=2) 

x.shape =(10,100)