mindspore.amp.init_status

mindspore.amp.init_status()[source]

Returns a Tensor indicating initialized status for overflow detection.

Note

Only Ascend need status to capture overflow status, you can also call this function on GPU or CPU, but the return value is useless.

Returns

Tensor, has the shape of (8,).

Supported Platforms:

Ascend GPU CPU

Examples

>>> status = amp.init_status()