该接口计划废弃,可以使用torch.nn.functional.mish接口进行替换。
torch_npu.npu_mish(self) -> Tensor
按元素计算self的双曲正切。
self (Tensor) - 数据类型:float16、float32。
1 2 3 4
>>> x = torch.rand(10, 30, 10).npu() >>> y = torch_npu.npu_mish(x) >>> y.shape torch.Size([10, 30, 10])