npu.ops.gelu

Applicability

Description

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

Prototype

1

npu.ops.gelu(x)

Parameters

Returns

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

Example

1
2

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