(beta)torch_npu.contrib.module.Mish
[!NOTICE]
该接口计划废弃,可以使用torch.nn.Mish接口进行替换。
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A3 训练系列产品 | √ |
| Atlas A2 训练系列产品 | √ |
| Atlas 训练系列产品 | √ |
| Atlas 推理系列产品 | √ |
功能说明
应用基于NPU的Mish操作。
函数原型
torch_npu.contrib.module.Mish(nn.Module)
调用示例
>>> import torch
>>> from torch_npu.contrib.module import Mish
>>> m = Mish()
>>> input_tensor = torch.randn(2, 32, 5, 5).npu()
>>> output = m(input_tensor)