Skip to contents

Ones shaped like an existing mlx array

Source: R/create.R
mlx_ones_like.Rd

mlx_ones_like() mirrors mlx.core.ones_like(), creating an array of ones with the same shape. Optionally override dtype.

Usage

mlx_ones_like(x, dtype = NULL)

Arguments

x

An mlx array.

dtype

Data type string. Supported types include:

  • Floating point: "float32", "float64"

  • Integer: "int8", "int16", "int32", "int64", "uint8", "uint16", "uint32", "uint64"

  • Other: "bool", "complex64"

Not all functions support all types. See individual function documentation.

Value

An mlx array of ones matching x.

Details

MLX does not support float64 operations on GPU. When this function creates a float64 array or converts one back to R, Rmlx temporarily switches only that internal creation or layout work to CPU. Later operations on the returned array still use the current mlx_device().

See also

mlx.core.ones_like

Examples

base <- mlx_full(c(2, 3), 5)
mlx_ones_like(base)
#> mlx array [2 x 3]
#>   dtype: float32
#>   values:
#>      [,1] [,2] [,3]
#> [1,]    1    1    1
#> [2,]    1    1    1