Python tf.math.reduce_max用法及代码示例

跨张量的维度计算元素的tf.math.maximum。

用法

tf.math.reduce_max(
    input_tensor, axis=None, keepdims=False, name=None
)

这是按元素tf.math.maximum op 的归约操作。

沿 axis 中给定的尺寸减少 input_tensor 。除非 keepdims 为真，否则对于 axis 中的每个条目，张量的秩都会减少 1，这必须是唯一的。如果 keepdims 为真，则保留缩减后的维度，长度为 1。

如果axis 为None，则所有维度都会减少，并返回具有单个元素的张量。

使用示例：

x = tf.constant([5, 1, 2, 4])
tf.reduce_max(x)
<tf.Tensor:shape=(), dtype=int32, numpy=5>
x = tf.constant([-5, -1, -2, -4])
tf.reduce_max(x)
<tf.Tensor:shape=(), dtype=int32, numpy=-1>
x = tf.constant([4, float('nan')])
tf.reduce_max(x)
<tf.Tensor:shape=(), dtype=float32, numpy=nan>
x = tf.constant([float('nan'), float('nan')])
tf.reduce_max(x)
<tf.Tensor:shape=(), dtype=float32, numpy=nan>
x = tf.constant([float('-inf'), float('inf')])
tf.reduce_max(x)
<tf.Tensor:shape=(), dtype=float32, numpy=inf>

请参阅 np.amax 和 np.nanmax 行为的 numpy 文档。