check_numerics

用来调试一个 Tensor ，计算并返回这个 Tensor 数值中异常值(NaN、Inf) 和零元素的数量。

参数

tensor (Tensor) – 需要检查的目标 Tensor 。
op_type (str) – 产生目标 Tensor 的 OP 或 API 。
var_name (str) – 目标 Tensor 的名字。
debug_mode (paddle.amp.debugging.DebugMode, 可选) - 要使用的调试类型。默认值为 paddle.amp.debugging.DebugMode.CHECK_NAN_INF_AND_ABORT。

返回

stats(Tensor)，保存目标 Tensor 统计信息的 Tensor ，形状为 [3]，依次存放目标 Tensor 中 NaN、Inf 和零元素的数量。数据类型为 int64。
values(Tensor)，保存目标 Tensor 的最大值、最小值和所有元素的均值，形状为 [3]，数据类型为 float 。

代码示例

COPY-FROM: paddle.amp.debugging.check_numerics