mindspore/docs/api/api_python/amp/mindspore.amp.all_finite.rst

18 lines
539 B
ReStructuredText
Raw Normal View History

2022-08-30 16:52:05 +08:00
mindspore.amp.all_finite
========================
2023-01-03 12:11:23 +08:00
.. py:function:: mindspore.amp.all_finite(inputs, status=None)
2022-08-30 16:52:05 +08:00
检查inputs是否是有效值无溢出
.. note::
这是一个实验性接口,后续可能删除或修改。
此接口只在整网训练情况下用于判断梯度是否溢出,返回结果在不同后端可能存在差异。
参数:
- **inputs** (Union(tuple(Tensor), list(Tensor))) - 可迭代的Tensor。
返回:
2022-09-23 15:58:51 +08:00
Tensor布尔类型的标量Tensor。