Tensorflow

TensorFlow学习笔记(1)tf.reduce_mean,

2018-11-09  本文已影响0人  谢昆明
    import tensorflow as tf
    import numpy as np

    #  Computes the mean of elements across dimensions of a tensor. (deprecated arguments)
    
    #  tf.reduce_mean(
    #      input_tensor,
    #      axis=None,
    #      keepdims=None,
    #      name=None,
    #      reduction_indices=None,
    #      keep_dims=None
    #      )
    
    c = np.array([[3.,4], [5.,6], [6.,7]])
    
    step = tf.reduce_mean(c, 1)
    with tf.Session() as sess:
        print(sess.run(step))

输出

[3.5 5.5 6.5]

也就是
[3.,4]的平均值是3.5,类推

用例对应的源代码,觉得有帮助可以Star

上一篇 下一篇

猜你喜欢

热点阅读