Cross
Returns the cross product of vectors in dimension dim of input and other.
Abstract Signature:
Cross(input: Tensor, other: Tensor, dim: int)
PyTorch
API: torch.cross
Strategy: Direct Mapping
JAX (Core)
API: jax.numpy.cross
Strategy: Direct Mapping
NumPy
API: numpy.cross
Strategy: Direct Mapping
Keras
API: keras.ops.cross
Strategy: Direct Mapping
TensorFlow
API: tf.linalg.cross
Strategy: Direct Mapping
Apple MLX
API: mlx.core.cross
Strategy: Direct Mapping
Flax NNX
API: jax.numpy.cross
Strategy: Direct Mapping
PaxML / Praxis
API: jax.numpy.cross
Strategy: Direct Mapping