ai.onnx.LpNormalization

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

Description

Applies Lp-normalization to the input tensor along the specified axis: output = input / Lp_norm(input, axis). Supports L1 (p=1) and L2 (p=2) norms. Where the Lp norm is zero, the output is defined as zero to avoid division by zero.

See the ONNX LpNormalization spec for the reference semantics.

Inputs

Name Bind key Logical dtype Rank Shape Description Presence
input input T Input tensor to normalize. required

Outputs

Name Bind key Logical dtype Rank Shape Description Presence
output output T same as input same as input Tensor after Lp-normalization; same shape as the input. required

Attributes

Default values (overridable per request):

Attribute Default Description
axis -1 The axis along which normalization is applied; -1 means the last axis.
p 2 The order of the Lp norm to use; only 1 (L1) or 2 (L2) are supported.

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.LpNormalization", { version: 1 });
const { output } = await kernel({ input: { data: inputData, shape: [2, 2] } });
Downloads last month
-
kernel
webgpu
wgsl
apache-2.0
WebGPU

Requires WebGPU support. See the compatibility table.