mindspore.amp.all_finite

mindspore.amp.all_finite(inputs, status=None)[源代码]

检查inputs是否是有效值（无溢出）。

说明

这是一个实验性接口，后续可能删除或修改。

此接口只在整网训练情况下用于判断梯度是否溢出，返回结果在不同后端可能存在差异。

参数：

  • inputs (Union(tuple(Tensor), list(Tensor))) - 可迭代的Tensor。

  • status (Tensor) - 溢出检测时所需要的初始状态，仅在Ascend需要。默认值：None。

返回：

Tensor，布尔类型的标量Tensor。

支持平台：

Ascend GPU CPU

样例：

>>> x = (Tensor(np.array([np.log(-1), 1, np.log(0)])), Tensor(np.array([1.0]))
>>> output = amp.all_finite(x)