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()