Int16Tensor

Constructor/Type for 16-bit integer tensors.

Abstract Signature:

Int16Tensor(data)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.ShortTensor

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro 'jnp.array({data}, dtype=jnp.int16)'

NumPy

API: —

Strategy: Macro 'np.array({data}, dtype=np.int16)'

Keras

API: —

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

TensorFlow

API: —

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

Apple MLX

API: —

Strategy: Macro 'mx.array({data}, dtype=mx.int16)'

Flax NNX

API: —

Strategy: Macro 'jnp.array({data}, dtype=jnp.int16)'

PaxML / Praxis

API: —

Strategy: Macro 'jnp.array({data}, dtype=jnp.int16)'