Vector cross product with mlx arrays
Usage
mlx_cross(a, b, axis = NULL)
Arguments
- a, b
Input mlx arrays containing 3D vectors.
- axis
Axis along which to compute the cross product (1-indexed).
Omit the argument to use the trailing dimension.
Value
An mlx array of cross products.
Examples
u <- as_mlx(c(1, 0, 0))
v <- as_mlx(c(0, 1, 0))
mlx_cross(u, v)
#> mlx array [3]
#> dtype: float32
#> device: gpu
#> values:
#> [1] 0 0 1