ai.onnx.Max
ai.onnx · standard ONNX operator · ONNX opset ≥ 13
Description
Computes the elementwise maximum across one or more input tensors with NumPy-style multidirectional broadcasting. All inputs must share the same data type, and the output has the broadcasted shape.
See the ONNX Max spec for the reference semantics.
Inputs
| Name | Bind key | Logical dtype | Rank | Shape | Description | Presence |
|---|---|---|---|---|---|---|
A |
a |
T |
— | — | First input tensor. | required |
B |
b |
T |
— | — | Second input tensor, broadcast-compatible with A. | optional |
C |
c |
T |
— | — | Third input tensor, broadcast-compatible with A and B. | optional |
D |
d |
T |
— | — | Fourth input tensor, broadcast-compatible with all other inputs. | optional |
E |
e |
T |
— | — | Fifth input tensor, broadcast-compatible with all other inputs. | optional |
Outputs
| Name | Bind key | Logical dtype | Rank | Shape | Description | Presence |
|---|---|---|---|---|---|---|
max |
y |
T |
derived | derived; see description | Elementwise maximum of all input tensors. | required |
Type constraints
| Variable | Allowed dtypes |
|---|---|
T |
float32, float16, int32, uint32, int16, int8, uint8 |
Files
metadata.json— kernel metadata (id, digests, provenance)manifest.json— the op contract (source of truth)test.json— correctness casesbench.json— benchmark + tuning casesdatamove-elementwise-copy.wgsl.jinjaminmax-broadcast.wgsl.jinjaminmax-vec4.wgsl.jinja
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.Max", { version: 1 });
const { y } = await kernel({ a: { data: aData, shape: [] } });
- Downloads last month
- -
kernel
webgpu
wgsl
apache-2.0
WebGPU
Requires WebGPU support. See the compatibility table.