mindspore.amp.all_finite

查看源文件
mindspore.amp.all_finite(inputs)[源代码]

检查inputs是否是有效值(无溢出)。

警告

这是一个实验性API,后续可能修改或删除。

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

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

返回:

Tensor,布尔类型的标量Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import amp, Tensor
>>> import numpy as np
>>> x = (Tensor(np.array([np.log(-1), 1, np.log(0)])), Tensor(np.array([1.0])))
>>> output = amp.all_finite(x)
教程样例: