(beta)torch_npu.npu_mish
该接口计划废弃,可以使用torch.nn.functional.mish接口进行替换。
接口原型
torch_npu.npu_mish(self) -> Tensor
功能描述
按元素计算self的双曲正切。
参数说明
self (Tensor) - 数据类型:float16、float32。
调用示例
>>> x = torch.rand(10, 30, 10).npu() >>> y = torch_npu.npu_mish(x) >>> y.shape torch.Size([10, 30, 10])
父主题: torch_npu