2023-11-29 12:41:56 -08:00
|
|
|
.. _ops:
|
|
|
|
|
|
|
|
|
|
Operations
|
|
|
|
|
==========
|
|
|
|
|
|
|
|
|
|
.. currentmodule:: mlx.core
|
|
|
|
|
|
2024-04-11 17:33:33 -07:00
|
|
|
.. autosummary::
|
2023-11-29 12:41:56 -08:00
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
|
|
abs
|
|
|
|
|
add
|
2024-05-20 09:40:17 -07:00
|
|
|
addmm
|
2023-11-29 12:41:56 -08:00
|
|
|
all
|
2024-04-11 17:33:33 -07:00
|
|
|
allclose
|
2023-11-29 12:41:56 -08:00
|
|
|
any
|
|
|
|
|
arange
|
|
|
|
|
arccos
|
|
|
|
|
arccosh
|
|
|
|
|
arcsin
|
|
|
|
|
arcsinh
|
|
|
|
|
arctan
|
2024-05-10 08:49:36 -07:00
|
|
|
arctan2
|
2023-11-29 12:41:56 -08:00
|
|
|
arctanh
|
|
|
|
|
argmax
|
|
|
|
|
argmin
|
|
|
|
|
argpartition
|
|
|
|
|
argsort
|
|
|
|
|
array_equal
|
2024-05-20 09:40:17 -07:00
|
|
|
as_strided
|
2024-02-29 12:39:18 -08:00
|
|
|
atleast_1d
|
|
|
|
|
atleast_2d
|
|
|
|
|
atleast_3d
|
2024-05-10 08:49:36 -07:00
|
|
|
bitwise_and
|
|
|
|
|
bitwise_or
|
|
|
|
|
bitwise_xor
|
2024-04-26 08:24:09 -07:00
|
|
|
block_masked_mm
|
2024-05-10 08:49:36 -07:00
|
|
|
broadcast_to
|
2023-12-17 13:23:03 -08:00
|
|
|
ceil
|
2023-12-21 22:13:41 -08:00
|
|
|
clip
|
2023-11-29 12:41:56 -08:00
|
|
|
concatenate
|
2024-05-10 08:49:36 -07:00
|
|
|
conj
|
|
|
|
|
conjugate
|
2023-11-29 12:41:56 -08:00
|
|
|
convolve
|
|
|
|
|
conv1d
|
|
|
|
|
conv2d
|
2024-08-23 12:14:53 -07:00
|
|
|
conv3d
|
2024-09-17 12:06:14 -07:00
|
|
|
conv_transpose1d
|
|
|
|
|
conv_transpose2d
|
|
|
|
|
conv_transpose3d
|
2024-02-29 12:39:18 -08:00
|
|
|
conv_general
|
2023-11-29 12:41:56 -08:00
|
|
|
cos
|
|
|
|
|
cosh
|
2024-03-30 17:32:20 -07:00
|
|
|
cummax
|
|
|
|
|
cummin
|
|
|
|
|
cumprod
|
|
|
|
|
cumsum
|
2024-04-26 08:24:09 -07:00
|
|
|
degrees
|
2023-12-21 22:13:41 -08:00
|
|
|
dequantize
|
2024-02-01 13:08:29 -08:00
|
|
|
diag
|
|
|
|
|
diagonal
|
2023-11-29 12:41:56 -08:00
|
|
|
divide
|
2024-01-10 14:14:38 -08:00
|
|
|
divmod
|
2024-07-25 11:59:11 -07:00
|
|
|
einsum
|
|
|
|
|
einsum_path
|
2023-11-29 12:41:56 -08:00
|
|
|
equal
|
|
|
|
|
erf
|
|
|
|
|
erfinv
|
|
|
|
|
exp
|
2024-04-11 17:33:33 -07:00
|
|
|
expm1
|
2023-11-29 12:41:56 -08:00
|
|
|
expand_dims
|
2023-12-13 14:46:24 -08:00
|
|
|
eye
|
2023-12-17 13:23:03 -08:00
|
|
|
flatten
|
2023-12-21 22:13:41 -08:00
|
|
|
floor
|
|
|
|
|
floor_divide
|
2023-11-29 12:41:56 -08:00
|
|
|
full
|
2024-05-23 21:11:24 -07:00
|
|
|
gather_mm
|
|
|
|
|
gather_qmm
|
2023-11-29 12:41:56 -08:00
|
|
|
greater
|
|
|
|
|
greater_equal
|
2024-07-11 15:32:08 -07:00
|
|
|
hadamard_transform
|
2023-12-13 14:46:24 -08:00
|
|
|
identity
|
2024-01-10 14:14:38 -08:00
|
|
|
inner
|
2024-09-17 12:06:14 -07:00
|
|
|
isfinite
|
2024-03-14 12:46:45 -07:00
|
|
|
isclose
|
2024-03-30 17:32:20 -07:00
|
|
|
isinf
|
2024-01-17 17:15:29 -08:00
|
|
|
isnan
|
|
|
|
|
isneginf
|
2024-03-30 17:32:20 -07:00
|
|
|
isposinf
|
2024-05-20 09:40:17 -07:00
|
|
|
issubdtype
|
2024-05-10 08:49:36 -07:00
|
|
|
left_shift
|
2023-11-29 12:41:56 -08:00
|
|
|
less
|
|
|
|
|
less_equal
|
2023-12-21 22:13:41 -08:00
|
|
|
linspace
|
2023-11-29 12:41:56 -08:00
|
|
|
load
|
|
|
|
|
log
|
|
|
|
|
log2
|
|
|
|
|
log10
|
|
|
|
|
log1p
|
|
|
|
|
logaddexp
|
|
|
|
|
logical_not
|
2024-01-10 14:14:38 -08:00
|
|
|
logical_and
|
|
|
|
|
logical_or
|
2023-11-29 12:41:56 -08:00
|
|
|
logsumexp
|
|
|
|
|
matmul
|
|
|
|
|
max
|
|
|
|
|
maximum
|
|
|
|
|
mean
|
2024-04-11 17:33:33 -07:00
|
|
|
meshgrid
|
2023-11-29 12:41:56 -08:00
|
|
|
min
|
|
|
|
|
minimum
|
2023-12-17 13:23:03 -08:00
|
|
|
moveaxis
|
2023-11-29 12:41:56 -08:00
|
|
|
multiply
|
2024-07-25 11:59:11 -07:00
|
|
|
nan_to_num
|
2023-11-29 12:41:56 -08:00
|
|
|
negative
|
2024-04-26 08:24:09 -07:00
|
|
|
not_equal
|
2023-11-29 12:41:56 -08:00
|
|
|
ones
|
|
|
|
|
ones_like
|
2024-01-10 14:14:38 -08:00
|
|
|
outer
|
2023-11-29 12:41:56 -08:00
|
|
|
partition
|
|
|
|
|
pad
|
2024-05-20 09:40:17 -07:00
|
|
|
power
|
2023-11-29 12:41:56 -08:00
|
|
|
prod
|
2024-09-28 11:04:59 -07:00
|
|
|
put_along_axis
|
2023-12-21 22:13:41 -08:00
|
|
|
quantize
|
|
|
|
|
quantized_matmul
|
2024-04-26 08:24:09 -07:00
|
|
|
radians
|
2023-11-29 12:41:56 -08:00
|
|
|
reciprocal
|
2024-05-20 09:40:17 -07:00
|
|
|
remainder
|
2024-01-03 20:14:05 -08:00
|
|
|
repeat
|
2023-11-29 12:41:56 -08:00
|
|
|
reshape
|
2024-05-10 08:49:36 -07:00
|
|
|
right_shift
|
2023-12-21 22:13:41 -08:00
|
|
|
round
|
2023-11-29 12:41:56 -08:00
|
|
|
rsqrt
|
|
|
|
|
save
|
|
|
|
|
savez
|
|
|
|
|
savez_compressed
|
2024-01-10 14:14:38 -08:00
|
|
|
save_gguf
|
2024-01-03 20:14:05 -08:00
|
|
|
save_safetensors
|
2023-11-29 12:41:56 -08:00
|
|
|
sigmoid
|
|
|
|
|
sign
|
|
|
|
|
sin
|
|
|
|
|
sinh
|
|
|
|
|
softmax
|
|
|
|
|
sort
|
|
|
|
|
split
|
|
|
|
|
sqrt
|
|
|
|
|
square
|
|
|
|
|
squeeze
|
2023-12-17 13:23:03 -08:00
|
|
|
stack
|
2024-04-11 17:33:33 -07:00
|
|
|
std
|
2023-11-29 12:41:56 -08:00
|
|
|
stop_gradient
|
|
|
|
|
subtract
|
|
|
|
|
sum
|
2023-12-17 13:23:03 -08:00
|
|
|
swapaxes
|
2023-11-29 12:41:56 -08:00
|
|
|
take
|
|
|
|
|
take_along_axis
|
|
|
|
|
tan
|
|
|
|
|
tanh
|
2024-01-03 20:14:05 -08:00
|
|
|
tensordot
|
2024-03-14 12:46:45 -07:00
|
|
|
tile
|
|
|
|
|
topk
|
2024-05-23 21:11:24 -07:00
|
|
|
trace
|
2023-11-29 12:41:56 -08:00
|
|
|
transpose
|
2023-12-17 13:23:03 -08:00
|
|
|
tri
|
|
|
|
|
tril
|
|
|
|
|
triu
|
2023-11-29 12:41:56 -08:00
|
|
|
var
|
2024-06-06 20:28:06 -07:00
|
|
|
view
|
2023-11-29 12:41:56 -08:00
|
|
|
where
|
|
|
|
|
zeros
|
|
|
|
|
zeros_like
|
