（beta）torch_npu.contrib.module.Mish
该接口计划废弃，可以使用torch.nn.Mish接口进行替换。
接口原型
torch_npu.contrib.module.Mish(nn.Module):
功能描述
应用基于NPU的Mish操作。
支持的型号
Atlas 训练系列产品 Atlas A2 训练系列产品 Atlas A3 训练系列产品 Atlas 推理系列产品
调用示例
1 2 3 4 | >>> from torch_npu.contrib.module import Mish >>> m = Mish() >>> input_tensor = torch.randn(2, 32, 5, 5).npu() >>> output = m(input_tensor) |
父主题： torch_npu.contrib