ai.onnx.LeakyRelu

ai.onnx · standard ONNX operator · ONNX opset ≥ 16

Description

Applies the leaky rectified linear unit activation elementwise: f(x) = x for x >= 0, and f(x) = alpha * x for x < 0. The output has the same shape and dtype as the input.

See the ONNX LeakyRelu spec for the reference semantics.

Inputs

Name Bind key Logical dtype Rank Shape Description Presence
X x T Values transformed elementwise by leaky ReLU. required

Outputs

Name Bind key Logical dtype Rank Shape Description Presence
Y y T same as X same as X Output tensor with the leaky ReLU activation applied elementwise. required

Attributes

Default values (overridable per request):

Attribute Default Description
alpha 0.01 Coefficient of leakage applied to negative input values; defaults to 0.01.

Type constraints

Variable Allowed dtypes
T float32, float16

Files

Use with @huggingface/kernels

The loader derives every required output's shape and logical dtype from the manifest contract and this call. It then allocates the result tensors automatically.

The version: 1 option selects the published kernel contract; it is independent of any operator opset, contrib since_version, or model version.

Replace each *Data placeholder with a typed array containing the corresponding input data.

import { getKernel } from "@huggingface/kernels";

const kernel = await getKernel("webgpu-kernels/ai.onnx.LeakyRelu", { version: 1 });
const { y } = await kernel({ x: { data: xData, shape: [5] } });
Downloads last month
-
kernel
webgpu
wgsl
apache-2.0
WebGPU

Requires WebGPU support. See the compatibility table.