（beta）torch_npu.contrib.Swish

[!NOTICE]
该接口计划废弃，可以使用torch_npu.contrib.SiLU接口进行替换。

产品支持情况

产品	是否支持
Atlas A3 训练系列产品	√
Atlas A2 训练系列产品	√
Atlas 训练系列产品	√
Atlas 推理系列产品	√

功能说明

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

函数原型

torch_npu.contrib.Swish()

调用示例

>>> import torch
>>> import torch_npu
>>> m = torch_npu.contrib.Swish().npu()
>>> input_tensor = torch.randn(2, 32, 5, 5).npu()
>>> output = m(input_tensor)