CastInt64

Casts tensor to 64-bit integer type.

Abstract Signature:

CastInt64(input: Tensor)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Macro '{input}.long()'

JAX (Core)

API: —

Strategy: Macro '{input}.astype(jax.numpy.int64)'

NumPy

API: —

Strategy: Macro '{input}.astype(numpy.int64)'

Keras

API: —

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

TensorFlow

API: —

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

Apple MLX

API: —

Strategy: Macro '{input}.astype(mlx.core.int64)'

Flax NNX

API: —

Strategy: Macro '{input}.astype(jax.numpy.int64)'

PaxML / Praxis

API: —

Strategy: Macro '{input}.astype(jax.numpy.int64)'