mindspore.ops.NPUAllocFloatStatus

class mindspore.ops.NPUAllocFloatStatus

Allocates a flag to store the overflow status.

The flag is a tensor whose shape is \((8,)\) and data type is mindspore.dtype.float32.

Note

Please refer to the Examples of mindspore.ops.NPUGetFloatStatus.

Outputs:

Tensor, has the shape of \((8,)\).

Supported Platforms:

Ascend

Examples

>>> alloc_status = ops.NPUAllocFloatStatus()
>>> output = alloc_status()
>>> print(output)
[0. 0. 0. 0. 0. 0. 0. 0.]