A series of tf.reduce

tf.reduce_max

参考文献

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
import numpy as np

a=np.array([[1, 2],
            [5, 3],
            [2, 6]])

b = tf.Variable(a)
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(b))
    print('************')
    # 对于二维矩阵，axis=0轴可以理解为列方向(矩阵向下的方向),axis=1轴可以理解为
    #行方向(矩阵向右方向)
    print(sess.run(tf.reduce_max(b, axis=1, keepdims=False)))  # keepdims=False,结果维度减小
    print('************')
    print(sess.run(tf.reduce_max(b, axis=1, keepdims=True))) # keepdims=Ture,结果维度不变
    print('************')
    print(sess.run(tf.reduce_max(b, axis=0, keepdims=True)))

[[1 2]
 [5 3]
 [2 6]]
************
[2 5 6]
************
[[2]
 [5]
 [6]]
************
[[5 6]]
************
[5 6]