torch_npu.contrib.Swish
API接口
torch_npu.contrib.Swish()
功能描述
应用基于NPU的Sigmoid线性单元（SiLU）函数，按元素方向。SiLU函数也称为swish函数。
示例
>>> m = nn.SiLU()
>>> input_tensor = torch.randn(2, 32, 5, 5)
>>> output = m(input_tensor)
父主题： torch_npu.contrib
torch_npu.contrib.Swish()
应用基于NPU的Sigmoid线性单元（SiLU）函数，按元素方向。SiLU函数也称为swish函数。
>>> m = nn.SiLU()
>>> input_tensor = torch.randn(2, 32, 5, 5)
>>> output = m(input_tensor)