（beta）torch_npu.npu_mish

接口原型

torch_npu.npu_mish(self) -> Tensor

功能描述

按元素计算self的双曲正切。

参数说明

self (Tensor) - 数据类型：float16、float32。

支持的型号

• Atlas 训练系列产品
• Atlas A2 训练系列产品
• Atlas A3 训练系列产品
• Atlas 推理系列产品

调用示例

1
2
3
4

>>> x = torch.rand(10, 30, 10).npu()
>>> y = torch_npu.npu_mish(x)
>>> y.shape
torch.Size([10, 30, 10])