mindspore.amp.all_finite
- mindspore.amp.all_finite(inputs, status=None)[源代码]
检查inputs是否是有效值(无溢出)。
Note
这是一个实验性接口,后续可能删除或修改。
此接口只在整网训练情况下用于判断梯度是否溢出,返回结果在不同后端可能存在差异。
- 参数:
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)