class Mish(nn.Module):
应用基于NPU的Mish操作。
无
from torch_npu.contrib.module import Mish m = nn.Mish()
>>> m = nn.Mish() >>> input_tensor = torch.randn(2, 32, 5, 5) >>> output = m(input_tensor)