ai.onnx.Sigmoid

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

Description

Applies the sigmoid function y = 1 / (1 + exp(-x)) elementwise to every value in the input tensor. The output has the same shape and dtype as the input, with all values mapped to the range (0, 1).

See the ONNX Sigmoid spec for the reference semantics.

Inputs

Name Bind key Logical dtype Rank Shape Description Presence
X x T Values mapped elementwise through the logistic sigmoid. 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 sigmoid applied elementwise. required

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

Requires WebGPU support. See the compatibility table.