Skip to contents

Create arrays of ones on MLX devices

Source: R/create.R
mlx_ones.Rd

Create arrays of ones on MLX devices

Usage

mlx_ones(
  dim,
  dtype = c("float32", "float64", "int8", "int16", "int32", "int64", "uint8", "uint16",
    "uint32", "uint64", "bool", "complex64"),
  device = mlx_default_device()
)

Arguments

dim

Integer vector specifying array dimensions (shape).

dtype

MLX dtype to use. One of "float32", "float64", "int8", "int16", "int32", "int64", "uint8", "uint16", "uint32", "uint64", "bool", or "complex64".

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 filled with ones.

See also

mlx.core.ones

Examples

ones <- mlx_ones(c(2, 2), dtype = "float64", device = "cpu")
ones_int <- mlx_ones(c(3, 3), dtype = "int32")