2017-06-17 82 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)