Create arrays of ones on MLX devices — mlx_ones • Rmlx

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 the array shape/dimensions.
dtype: MLX dtype to use. One of "float32", "float64", "int8", "int16", "int32", "int64", "uint8", "uint16", "uint32", "uint64", "bool", or "complex64".
device: Execution target: provide "gpu", "cpu", or an mlx_stream created via mlx_new_stream(). Defaults to the current mlx_default_device().

Value

An mlx array filled with ones.

See also

Examples

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