ai.onnx.ThresholdedRelu

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

Description

Applies the thresholded rectified linear unit elementwise: y = x if x > alpha, y = 0 otherwise. The threshold alpha defaults to 1.0.

See the ONNX ThresholdedRelu spec for the reference semantics.

Inputs

Name Bind key Logical dtype Rank Shape Description Presence
X x T Values passed through only when they are greater than alpha. required

Outputs

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

Attributes

Default values (overridable per request):

Attribute Default Description
alpha 1 Threshold value above which inputs pass through; elements at or below alpha are set to zero (default 1.0).

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.ThresholdedRelu", { version: 1 });
const { y } = await kernel({ x: { data: xData, shape: [6] } });
Downloads last month
-
kernel
webgpu
wgsl
apache-2.0
WebGPU

Requires WebGPU support. See the compatibility table.