(beta)torch_npu.contrib.module.SiLU

该接口计划废弃,可以使用torch.nn.SiLU接口进行替换。
接口原型
SiLU(nn.Module):
功能描述
按元素应用基于NPU的Sigmoid线性单元(SiLU)函数。SiLU函数也称为Swish函数。
调用示例
调用方式示例:
from torch_npu.contrib.module import SiLU m = nn.SiLU()
使用示例:
>>> m = nn.SiLU() >>> input_tensor = torch.randn(2, 32, 5, 5) >>> output = m(input_tensor)
父主题: torch_npu.contrib