如何在 tf.Tensor 中找到最大值？

【问题标题】：How do I find the max value in tf.Tensor?如何在 tf.Tensor 中找到最大值？
【发布时间】：2021-10-18 20:14:08
【问题描述】：

如何找到每个元素中的最大值以便得到 2、4、6、8？

import tensorflow as tf
a = tf.constant([
  [[1, 2]], [[3, 4]],
  [[5, 6]], [[7, 8]]])

我尝试了以下代码：

tf.reduce_max(a, keepdims=True)

但这只是给我 8 作为输出，而忽略其余部分。

【问题讨论】：

reduce_max 有一个 axis 参数，也许可以试试。

标签： python tensorflow tensorflow2.0

【解决方案1】：

您必须像这样将 axis 参数更改为 -1：

import tensorflow as tf
a = tf.constant([
  [[1, 2]], [[3, 4]],
  [[5, 6]], [[7, 8]]])

print(tf.reduce_max(a, axis=-1, keepdims=False))

'''
tf.Tensor(
[[2]
 [4]
 [6]
 [8]], shape=(4, 1), dtype=int32)
'''

因为您有一个 3D 张量并且想要访问最后一个维度。

【讨论】：