beta)torch_npu.contrib.module.Mish

该接口计划废弃,可以使用torch.nn.Mish接口进行替换。

接口原型

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)