-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))' –