Skip to contents

mlx_vector() is a convenience around mlx_array() for 1-D payloads.

Usage

mlx_vector(data, dtype = NULL, device = mlx_default_device())

Arguments

data

Atomic vector providing the elements (recycling is not allowed).

dtype

Optional MLX dtype. Defaults to "float32" for numeric input, "bool" for logical, and "complex64" for complex.

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 vector with dim = length(data).