当前位置:   article > 正文

torch.mean和tf.reduce_mean_torch.reducemean

torch.reducemean

x = tf.constant([[0.3776],
[0.3031],
[0.4818],
[0.3887]])
y=tf.reduce_mean(x) # 1.5
tf.reduce_mean(x, 0) # [1.5, 1.5]
tf.reduce_mean(x, 1) # [1., 2.]
with tf.Session() as sess:
print(sess.run(y))

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号