forked from mindspore-Ecosystem/mindspore
19 lines
654 B
ReStructuredText
19 lines
654 B
ReStructuredText
mindspore.amp.all_finite
|
||
========================
|
||
|
||
.. py:function:: mindspore.amp.all_finite(inputs, status=None)
|
||
|
||
检查inputs是否是有效值（无溢出）。
|
||
|
||
.. note::
|
||
这是一个实验性接口，后续可能删除或修改。
|
||
|
||
此接口只在整网训练情况下用于判断梯度是否溢出，返回结果在不同后端可能存在差异。
|
||
|
||
参数：
|
||
- **inputs** (Union(tuple(Tensor), list(Tensor))) - 可迭代的Tensor。
|
||
- **status** (Tensor) - 溢出检测时所需要的初始状态，仅在Ascend需要。默认值：None。
|
||
|
||
返回：
|
||
Tensor，布尔类型的标量Tensor。
|