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:
AscendGPUCPU
Examples
>>> status = amp.init_status()