Sigmoid

In-place sigmoid.

Abstract Signature:

Sigmoid_(input: Tensor)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.sigmoid_

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.nn.sigmoid

Strategy: Macro '{input} = jax.nn.sigmoid({input})'

NumPy

API: scipy.special.expit

Strategy: Macro '{input} = sp_special.expit({input})'

Keras

API: keras.ops.sigmoid

Strategy: Macro '{input} = keras.ops.sigmoid({input})'

TensorFlow

API: tf.math.sigmoid

Strategy: Macro '{input} = tf.math.sigmoid({input})'

Apple MLX

API: mlx.core.sigmoid

Strategy: Macro '{input} = mx.sigmoid({input})'

Flax NNX

API: jax.nn.sigmoid

Strategy: Macro '{input} = jax.nn.sigmoid({input})'

PaxML / Praxis

API: jax.nn.sigmoid

Strategy: Macro '{input} = jax.nn.sigmoid({input})'