Python - tensorflow.math.is_finite()

在 TensorFlow 中，tf.math.is_finite() 函数用来判断一个张量中的元素是否都是有限的数值，如实数或负数，而不是 NaN 或无穷大的值。本篇文章将会详细介绍 tf.math.is_finite() 函数的使用流程和示例。

语法

下面是 tf.math.is_finite() 函数的语法：

tf.math.is_finite(x, name=None)

其中，

• x：待检查的张量，类型为 Tensor；
• name：操作名称，类型为 str，默认为 None。

返回值

tf.math.is_finite() 函数返回一个布尔型的张量，其中每个元素的值为 True 或 False。

示例

下面的代码段展示了 tf.math.is_finite() 函数的用法：

import tensorflow as tf

# 检查一个张量是否有超限值
x = tf.constant([[1.0, 2.0], [3.0, float('inf')]])
is_finite = tf.math.is_finite(x)

print(is_finite.numpy()) # [[ True  True], [ True False]]

上述代码中，我们首先定义了一个 2x2 的张量 x，其中包含一个超限值 float('inf')。然后，我们调用 tf.math.is_finite() 函数来检查张量 x 中的每个元素是否都是实数或负数。

最后，打印输出 is_finite 张量的值，可以发现第二行第二列的元素是无穷大的，所以对应位置的值为 False，其余位置的值为 True。

注意事项

• tf.math.is_finite() 函数仅支持 float16，float32，float64，int32，int64，complex64 和 complex128 类型的张量。
• 如果待检查的张量中存在 NaN 或无穷大的元素，则输出的张量中对应元素的值为 False。
• tf.math.is_finite() 函数不改变待检查的张量，而是返回一个布尔型的新张量。

参考文献

• TensorFlow官方文档（v2.6.0）。tf.math.is_finite。