npu.ops.gelu

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	×
Atlas inference products	×
Atlas training products	√

Computes the Gaussian Error Linear Unit (GELU) activation function. Each input tensor is multiplied by one P(X <= x), where P(X) follows N(0, 1).

npu.ops.gelu(x)

Parameter	Input/Output	Description
x	Input	Input tensor of type float.

Result tensor after the GELU operation is performed on input x. The data type is the same as the input.

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