Row sums for mlx arrays

Row sums for mlx arrays

Usage

rowSums(x, ...)

# Default S3 method
rowSums(x, na.rm = FALSE, dims = 1, ...)

# S3 method for class 'mlx'
rowSums(x, na.rm = FALSE, dims = 1, ...)

Arguments

x

An array or mlx array.

...

Additional arguments forwarded to the base implementation.

na.rm

Logical; currently ignored for mlx arrays.

dims

Dimensions passed through to the base implementation when x is not an mlx array.

Value

An mlx array if x is_mlx, otherwise a numeric vector.

See also

mlx.core.sum

Examples

x <- mlx_matrix(1:6, 3, 2)
rowSums(x)
#> mlx array [3]
#>   dtype: float32
#>   device: gpu
#>   values:
#> [1] 5 7 9