ai.onnx.Mod
ai.onnx · standard ONNX operator · ONNX opset ≥ 13
Description
Performs elementwise binary modulo on tensors A and B with multidirectional broadcasting. When fmod is 0 (default), applies Python-style % with the sign of the divisor; when fmod is 1, applies C-style fmod with the sign of the dividend.
See the ONNX Mod spec for the reference semantics.
Inputs
| Name | Bind key | Logical dtype | Rank | Shape | Description | Presence |
|---|---|---|---|---|---|---|
A |
a |
T |
— | — | Dividend tensor. | required |
B |
b |
T |
— | — | Divisor tensor. | required |
Outputs
| Name | Bind key | Logical dtype | Rank | Shape | Description | Presence |
|---|---|---|---|---|---|---|
C |
c |
T |
derived | broadcast result of A and B |
Remainder tensor; same shape as the broadcast result of A and B. | required |
Attributes
Default values (overridable per request):
| Attribute | Default | Description |
|---|---|---|
fmod |
0 |
Controls the modulo mode: 0 (default) uses Python-style integer mod (sign of divisor); 1 uses C-style fmod (sign of dividend, floating-point types only). |
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 casesmod-vec4.wgsl.jinjamod.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.Mod", { version: 1 });
const { c } = await kernel({ a: { data: aData, shape: [3] }, b: { data: bData, shape: [3] } });
- Downloads last month
- -
Requires WebGPU support. See the compatibility table.