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 or device.

Usage

mlx_ones_like(x, dtype = NULL, device = 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.

device

Execution target: supply "gpu", "cpu", or an mlx_stream created via mlx_new_stream(). Defaults to the current mlx_default_device() unless noted otherwise (helpers that act on an existing array typically reuse that array's device or stream).

Value

An mlx array of ones matching x.

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
#>   device: cpu
#>   values:
#>      [,1] [,2] [,3]
#> [1,]    1    1    1
#> [2,]    1    1    1