该接口计划废弃,可以使用torch_npu.contrib.ModulationDeformCon接口进行替换。
torch_npu.contrib.Swish()
应用基于NPU的Sigmoid线性单元(SiLU)函数,按元素方向。SiLU函数也称为swish函数。
1 2 3 | >>> m = torch_npu.contrib.Swish().npu() >>> input_tensor = torch.randn(2, 32, 5, 5).npu() >>> output = m(input_tensor) |