check_numerics

paddle.amp.debugging. check_numerics ( tensor, op_type, var_name, debug_mode=paddle.amp.debugging.DebugMode.CHECK_NAN_INF_AND_ABORT )

用来调试一个 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