文档
注册

npu.ops.gelu

功能说明

计算高斯误差线性单元 (GELU) 激活函数。将输入Tensor乘以1个P(X <= x),其中P(X) ~ N(0, 1)。

函数原型

npu.ops.gelu (x)

参数说明

参数名

输入/输出

描述

x

输入

输入Tensor,float类型。

返回值

tensor:对输入x执行完GELU操作之后的输出Tensor。数据类型和输入相同。

调用示例

1
2
import npu_device as npu
output = npu.ops.gelu(x)
搜索结果
找到“0”个结果

当前产品无相关内容

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