npu.ops.gelu

函数原型

npu.ops.gelu (x)

功能说明

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

参数说明

返回值

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

调用示例

1
2

import npu_device as npu
output = npu.ops.gelu(x)