torch_npu.contrib.Swish

torch_npu.contrib.Swish()

应用基于NPU的Sigmoid线性单元（SiLU）函数，按元素方向。SiLU函数也称为swish函数。

• 参数解释：

  无参数。

• 约束条件：

  无

• 示例：

      >>> m = nn.SiLU()
      >>> input_tensor = torch.randn(2, 32, 5, 5)
      >>> output = m(input_tensor)