Python - tensorflow.math.maximum()

简介

• tensorflow.math.maximum() 是 TensorFlow 中的一个数学库函数，用于获取两个 tensor 的逐元素最大值。

语法

tensorflow.math.maximum(x, y, name=None)

参数

• x 和 y 是形状相同的张量（tensor），可以是任何类型，包括整数和浮点数。

返回值

• 返回一个新的张量，其形状与输入的张量相同，每个元素为 x 和 y 相应位置元素的最大值。

实例

import tensorflow as tf

a = tf.constant([1, 2, 3, 4])
b = tf.constant([3, 1, 5, 0])

max_values = tf.math.maximum(a, b)

print(max_values.numpy())

输出如下：

[3 2 5 4]

总结

• tensorflow.math.maximum() 函数非常简单，但在机器学习的许多应用中可能很有用。您可以使用它来比较两个 tensor，并进行逐元素的最大值计算。