(beta)torch_npu.npu_clear_float_status

产品支持情况

产品 是否支持
Atlas A3 训练系列产品
Atlas A2 训练系列产品
Atlas 推理系列产品
Atlas 训练系列产品

功能说明

清除溢出检测相关标志位。

函数原型

torch_npu.npu_clear_float_status(self) -> Tensor

参数说明

self(Tensor):数据类型为float32的张量。

返回值说明

Tensor

一个包含8个float32类型全零值的Tensor。

调用示例

>>> import torch, torch_npu
>>> x = torch.rand(2).npu()
>>> torch_npu.npu_clear_float_status(x)
tensor([0., 0., 0., 0., 0., 0., 0., 0.], device='npu:0')