2022-08-26 15:31:01 +08:00
|
|
|
|
mindspore.ops.NPUGetFloatStatus
|
|
|
|
|
|
================================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:class:: mindspore.ops.NPUGetFloatStatus
|
|
|
|
|
|
|
2022-12-21 12:44:28 +08:00
|
|
|
|
在执行 :class:`mindspore.ops.NPUAllocFloatStatus` 后， `NPUGetFloatStatus` 获取最新溢出状态并更新标识。
|
2022-08-26 15:31:01 +08:00
|
|
|
|
|
2022-11-02 19:28:41 +08:00
|
|
|
|
.. note::
|
2022-12-21 12:44:28 +08:00
|
|
|
|
标志是一个Tensor，其shape为 :math:`(8,)` ，数据类型为 `mindspore.dtype.float32` 。如果标志的和等于0，则没有发生溢出。如果标志之和大于0，则发生溢出。此外，使用有严格的顺序要求，即在使用 `NPUGetFloatStatus` 算子之前，需要确保 :class:`mindspore.ops.NPUClearFloatStatus` 和需执行的计算已执行。使用 :class:`mindspore.ops.Depend` 确保执行顺序。
|
2022-08-26 15:31:01 +08:00
|
|
|
|
|
|
|
|
|
|
输入：
|
2022-12-21 12:44:28 +08:00
|
|
|
|
- **x** (Tensor) - :class:`mindspore.ops.NPUAllocFloatStatus` 的输出Tensor。数据类型必须为float16或float32。 :math:`(N,*)` ，其中 :math:`*` 表示任意附加维度，其rank应小于8。
|
2022-08-26 15:31:01 +08:00
|
|
|
|
|
|
|
|
|
|
输出：
|
|
|
|
|
|
Tensor，shape与 `x` 相同。Tensor中的所有元素都将为零。
|
|
|
|
|
|
|
|
|
|
|
|
异常：
|
|
|
|
|
|
- **TypeError** - 如果 `x` 不是Tensor。
|
|
|
|
|
|
- **TypeError** - 如果 `x` 的数据类型既不是float16也不是float32。
|
