当前位置:   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博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/133165
推荐阅读
相关标签
  

闽ICP备14008679号