mlx.core.array#

class array#

An N-dimensional array object.

__init__(self: array, val: scalar | list | tuple | ndarray | array, dtype: Dtype | None = None)#

Methods

__init__(self, val[, dtype])

abs(self, *[, stream])

See abs().

all(self[, axis, keepdims, stream])

See all().

any(self[, axis, keepdims, stream])

See any().

argmax(self[, axis, keepdims, stream])

See argmax().

argmin(self[, axis, keepdims, stream])

See argmin().

astype(self, dtype[, stream])

Cast the array to a specified type.

cos(self, *[, stream])

See cos().

cummax(self[, axis, reverse, inclusive, stream])

See cummax().

cummin(self[, axis, reverse, inclusive, stream])

See cummin().

cumprod(self[, axis, reverse, inclusive, stream])

See cumprod().

cumsum(self[, axis, reverse, inclusive, stream])

See cumsum().

diag(self[, k, stream])

Extract a diagonal or construct a diagonal matrix.

diagonal(self[, offset, axis1, axis2, stream])

See diagonal().

exp(self, *[, stream])

See exp().

flatten(self[, start_axis, end_axis, stream])

See flatten().

item(self)

Access the value of a scalar array.

log(self, *[, stream])

See log().

log10(self, *[, stream])

See log10().

log1p(self, *[, stream])

See log1p().

log2(self, *[, stream])

See log2().

logsumexp(self[, axis, keepdims, stream])

See logsumexp().

max(self[, axis, keepdims, stream])

See max().

mean(self[, axis, keepdims, stream])

See mean().

min(self[, axis, keepdims, stream])

See min().

moveaxis(self, source, destination, *[, stream])

See moveaxis().

prod(self[, axis, keepdims, stream])

See prod().

reciprocal(self, *[, stream])

See reciprocal().

reshape(self, *shape[, stream])

Equivalent to reshape() but the shape can be passed either as a tuple or as separate arguments.

round(self[, decimals, stream])

See round().

rsqrt(self, *[, stream])

See rsqrt().

sin(self, *[, stream])

See sin().

split(self, indices_or_sections[, axis, stream])

See split().

sqrt(self, *[, stream])

See sqrt().

square(self, *[, stream])

See square().

squeeze(self[, axis, stream])

See squeeze().

sum(self[, axis, keepdims, stream])

See sum().

swapaxes(self, axis1, axis2, *[, stream])

See swapaxes().

tolist(self)

Convert the array to a Python list.

transpose(self, *axes[, stream])

Equivalent to transpose() but the axes can be passed either as a tuple or as separate arguments.

var(self[, axis, keepdims, ddof, stream])

See var().

Attributes

T

Equivalent to calling self.transpose() with no arguments.

at

Used to apply updates at the given indices.

dtype

The array's Dtype.

itemsize

The size of the array's datatype in bytes.

nbytes

The number of bytes in the array.

ndim

The array's dimension.

shape

The shape of the array as a Python tuple.

size

Number of elements in the array.