如何从两个张量得到均方？

Question

我有两个形状(10, 100, 2)的张量 如何获得形状(10, 100)的均方损失？

以下是我的尝试：

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

Answer 1

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

这实现了均方的公式：l2_loss（targets-logits）

Answer 2

只需使用axis的{​​{1}}参数：

tf.reduce_mean

x.shape =（10,100）