Python tf.raw_ops.CheckNumerics用法及代码示例

检查张量的 NaN 和 Inf 值。

用法

tf.raw_ops.CheckNumerics(
    tensor, message, name=None
)

参数

tensor 一个Tensor。必须是以下类型之一：bfloat16 , half , float32 , float64。
message 一个string。错误消息的前缀。
name 操作的名称(可选)。

一个Tensor。具有与 tensor 相同的类型。

运行时，如果 tensor 具有任何不是数字 (NaN) 或无穷大 (Inf) 的值，则报告 InvalidArgument 错误。否则，返回输入张量。

示例用法：

a = tf.Variable(1.0)
tf.debugging.check_numerics(a, message='')

b = tf.Variable(np.nan)
try:
  tf.debugging.check_numerics(b, message='Checking b')
except Exception as e:
  assert "Checking b:Tensor had NaN values" in e.message

c = tf.Variable(np.inf)
try:
  tf.debugging.check_numerics(c, message='Checking c')
except Exception as e:
  assert "Checking c:Tensor had Inf values" in e.message

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.raw_ops.CheckNumerics。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。