文档
注册

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')
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词