Short

Casts the input tensor to int16 (short).

Abstract Signature:

Short(input: Tensor)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.short

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.numpy.astype

Strategy: Macro '{input}.astype(jnp.int16)'

NumPy

API: numpy.astype

Strategy: Macro '{input}.astype(np.int16)'

Keras

API: keras.ops.cast

Strategy: Macro 'keras.ops.cast({input}, 'int16')'

TensorFlow

API: tf.cast

Strategy: Macro 'tf.cast({input}, tf.int16)'

Apple MLX

API: mlx.core.astype

Strategy: Macro '{input}.astype(mx.int16)'

Flax NNX

API: jax.numpy.astype

Strategy: Macro '{input}.astype(jnp.int16)'

PaxML / Praxis

API: jax.numpy.astype

Strategy: Macro '{input}.astype(jnp.int16)'