话不多说,直接上示例,就看懂了
import tensorflow as tf
max_value = tf.reduce_max([[1, 3, 2],[4,5,6]],reduction_indices=[1])
with tf.Session() as sess:
max_value = sess.run(max_value)
print(max_value)
# 输出
[3 6]
import tensorflow as tf
max_value = tf.reduce_max([[1, 3, 2],[4,5,6]],reduction_indices=[0])
with tf.Session() as sess:
max_value = sess.run(max_value)
print(max_value)
# 输出
[4,5,6]
通过以上例子,可以看出,该函数是找出列表或数组行上或者列上的最大值,并输出值,单个值直接输出,多个组成一个列表输出。参数reduction_indices=[]其中1表示取每行最大值输出,0表示取每列最大值输出。