tensorflow相关函数___tf.reduce_max()

80 阅读 0 评论 53 点赞

我是靠谱客的博主留胡子日记本，最近开发中收集的这篇文章主要介绍tensorflow相关函数___tf.reduce_max()，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

1、找出最大值

import tensorflow as tf
max_value = tf.reduce_max([1, 3, 2])
with tf.Session() as sess:
    max_value = sess.run(max_value)
    print(max_value)

输出结果：

2、找出列上的最大值

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]

3、找出行上的最大值

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]

注意：行是reduction_indices=[1] ；