class SiLU()

API接口

class 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)