(beta)torch_npu.npu_silu
该接口计划废弃,可以使用torch.nn.functional.silu接口进行替换。
接口原型
torch_npu.npu_silu(self) -> Tensor
功能描述
计算self的Swish。
参数说明
self (Tensor) - 数据类型:float16、float32
调用示例
>>> a=torch.rand(2,8).npu() >>> output = torch_npu.npu_silu(a) >>> output tensor([[0.4397, 0.7178, 0.5190, 0.2654, 0.2230, 0.2674, 0.6051, 0.3522], [0.4679, 0.1764, 0.6650, 0.3175, 0.0530, 0.4787, 0.5621, 0.4026]], device='npu:0')
父主题: torch_npu