Module ops 

Operations

Modules

indexing

Indexing Arrays

windows

Window functions for signal processing.

Macros

absDeprecated

Macro generated for the function crate::ops::abs. See the function documentation for more details.

acosDeprecated

Macro generated for the function crate::ops::acos. See the function documentation for more details.

acoshDeprecated

Macro generated for the function crate::ops::acosh. See the function documentation for more details.

addDeprecated

Macro generated for the function crate::ops::add. See the function documentation for more details.

addmmDeprecated

Macro generated for the function crate::ops::addmm. See the function documentation for more details.

allDeprecated

Macro generated for the function crate::ops::all. See the function documentation for more details.

all_axesDeprecated

Macro generated for the function crate::ops::all_axes. See the function documentation for more details.

all_axisDeprecated

Macro generated for the function crate::ops::all_axis. See the function documentation for more details.

all_closeDeprecated

Macro generated for the function crate::ops::all_close. See the function documentation for more details.

anyDeprecated

Macro generated for the function crate::ops::any. See the function documentation for more details.

any_axesDeprecated

Macro generated for the function crate::ops::any_axes. See the function documentation for more details.

any_axisDeprecated

Macro generated for the function crate::ops::any_axis. See the function documentation for more details.

arangeDeprecated

Macro generated for the function crate::ops::arange. See the function documentation for more details.

argpartitionDeprecated

Macro generated for the function crate::ops::argpartition. See the function documentation for more details.

argpartition_axisDeprecated

Macro generated for the function crate::ops::argpartition_axis. See the function documentation for more details.

argsortDeprecated

Macro generated for the function crate::ops::argsort. See the function documentation for more details.

argsort_axisDeprecated

Macro generated for the function crate::ops::argsort_axis. See the function documentation for more details.

array_eqDeprecated

Macro generated for the function crate::ops::array_eq. See the function documentation for more details.

as_stridedDeprecated

Macro generated for the function crate::ops::as_strided. See the function documentation for more details.

asinDeprecated

Macro generated for the function crate::ops::asin. See the function documentation for more details.

asinhDeprecated

Macro generated for the function crate::ops::asinh. See the function documentation for more details.

at_least_1dDeprecated

Macro generated for the function crate::ops::at_least_1d. See the function documentation for more details.

at_least_2dDeprecated

Macro generated for the function crate::ops::at_least_2d. See the function documentation for more details.

at_least_3dDeprecated

Macro generated for the function crate::ops::at_least_3d. See the function documentation for more details.

atanDeprecated

Macro generated for the function crate::ops::atan. See the function documentation for more details.

atan2Deprecated

Macro generated for the function crate::ops::atan2. See the function documentation for more details.

atanhDeprecated

Macro generated for the function crate::ops::atanh. See the function documentation for more details.

block_masked_mmDeprecated

Macro generated for the function crate::ops::block_masked_mm. See the function documentation for more details.

broadcast_arraysDeprecated

Macro generated for the function crate::ops::broadcast_arrays. See the function documentation for more details.

broadcast_toDeprecated

Macro generated for the function crate::ops::broadcast_to. See the function documentation for more details.

ceilDeprecated

Macro generated for the function crate::ops::ceil. See the function documentation for more details.

clipDeprecated

Macro generated for the function crate::ops::clip. See the function documentation for more details.

concatenateDeprecated

Macro generated for the function crate::ops::concatenate. See the function documentation for more details.

concatenate_axisDeprecated

Macro generated for the function crate::ops::concatenate_axis. See the function documentation for more details.

conv1dDeprecated

Macro generated for the function crate::ops::conv1d. See the function documentation for more details.

conv2dDeprecated

Macro generated for the function crate::ops::conv2d. See the function documentation for more details.

conv3dDeprecated

Macro generated for the function crate::ops::conv3d. See the function documentation for more details.

conv_generalDeprecated

Macro generated for the function crate::ops::conv_general. See the function documentation for more details.

conv_transpose1dDeprecated

Macro generated for the function crate::ops::conv_transpose1d. See the function documentation for more details.

conv_transpose2dDeprecated

Macro generated for the function crate::ops::conv_transpose2d. See the function documentation for more details.

conv_transpose3dDeprecated

Macro generated for the function crate::ops::conv_transpose3d. See the function documentation for more details.

cosDeprecated

Macro generated for the function crate::ops::cos. See the function documentation for more details.

coshDeprecated

Macro generated for the function crate::ops::cosh. See the function documentation for more details.

cummaxDeprecated

Macro generated for the function crate::ops::cummax. See the function documentation for more details.

cumminDeprecated

Macro generated for the function crate::ops::cummin. See the function documentation for more details.

cumprodDeprecated

Macro generated for the function crate::ops::cumprod. See the function documentation for more details.

cumsumDeprecated

Macro generated for the function crate::ops::cumsum. See the function documentation for more details.

degreesDeprecated

Macro generated for the function crate::ops::degrees. See the function documentation for more details.

dequantizeDeprecated

Macro generated for the function crate::ops::dequantize. See the function documentation for more details.

diagDeprecated

Macro generated for the function crate::ops::diag. See the function documentation for more details.

diagonalDeprecated

Macro generated for the function crate::ops::diagonal. See the function documentation for more details.

divideDeprecated

Macro generated for the function crate::ops::divide. See the function documentation for more details.

divmodDeprecated

Macro generated for the function crate::ops::divmod. See the function documentation for more details.

einsumDeprecated

Macro generated for the function crate::ops::einsum. See the function documentation for more details.

eqDeprecated

Macro generated for the function crate::ops::eq. See the function documentation for more details.

erfDeprecated

Macro generated for the function crate::ops::erf. See the function documentation for more details.

erfinvDeprecated

Macro generated for the function crate::ops::erfinv. See the function documentation for more details.

expDeprecated

Macro generated for the function crate::ops::exp. See the function documentation for more details.

expand_dimsDeprecated

Macro generated for the function crate::ops::expand_dims. See the function documentation for more details.

expand_dims_axesDeprecated

Macro generated for the function crate::ops::expand_dims_axes. See the function documentation for more details.

expm1Deprecated

Macro generated for the function crate::ops::expm1. See the function documentation for more details.

eyeDeprecated

Macro generated for the function crate::ops::eye. See the function documentation for more details.

flattenDeprecated

Macro generated for the function crate::ops::flatten. See the function documentation for more details.

floorDeprecated

Macro generated for the function crate::ops::floor. See the function documentation for more details.

floor_divideDeprecated

Macro generated for the function crate::ops::floor_divide. See the function documentation for more details.

from_fp8Deprecated

Macro generated for the function crate::ops::from_fp8. See the function documentation for more details.

fullDeprecated

Macro generated for the function crate::ops::full. See the function documentation for more details.

full_likeDeprecated

Macro generated for the function crate::ops::full_like. See the function documentation for more details.

gather_mmDeprecated

Macro generated for the function crate::ops::gather_mm. See the function documentation for more details.

gather_qmmDeprecated

Macro generated for the function crate::ops::gather_qmm. See the function documentation for more details.

geDeprecated

Macro generated for the function crate::ops::ge. See the function documentation for more details.

gtDeprecated

Macro generated for the function crate::ops::gt. See the function documentation for more details.

identityDeprecated

Macro generated for the function crate::ops::identity. See the function documentation for more details.

imagDeprecated

Macro generated for the function crate::ops::imag. See the function documentation for more details.

innerDeprecated

Macro generated for the function crate::ops::inner. See the function documentation for more details.

is_closeDeprecated

Macro generated for the function crate::ops::is_close. See the function documentation for more details.

is_infDeprecated

Macro generated for the function crate::ops::is_inf. See the function documentation for more details.

is_nanDeprecated

Macro generated for the function crate::ops::is_nan. See the function documentation for more details.

is_neg_infDeprecated

Macro generated for the function crate::ops::is_neg_inf. See the function documentation for more details.

is_pos_infDeprecated

Macro generated for the function crate::ops::is_pos_inf. See the function documentation for more details.

kronDeprecated

Macro generated for the function crate::ops::kron. See the function documentation for more details.

leDeprecated

Macro generated for the function crate::ops::le. See the function documentation for more details.

linspaceDeprecated

Macro generated for the function crate::ops::linspace. See the function documentation for more details.

logDeprecated

Macro generated for the function crate::ops::log. See the function documentation for more details.

log2Deprecated

Macro generated for the function crate::ops::log2. See the function documentation for more details.

log1pDeprecated

Macro generated for the function crate::ops::log1p. See the function documentation for more details.

log10Deprecated

Macro generated for the function crate::ops::log10. See the function documentation for more details.

logaddexpDeprecated

Macro generated for the function crate::ops::logaddexp. See the function documentation for more details.

logical_andDeprecated

Macro generated for the function crate::ops::logical_and. See the function documentation for more details.

logical_notDeprecated

Macro generated for the function crate::ops::logical_not. See the function documentation for more details.

logical_orDeprecated

Macro generated for the function crate::ops::logical_or. See the function documentation for more details.

logsumexpDeprecated

Macro generated for the function crate::ops::logsumexp. See the function documentation for more details.

logsumexp_axesDeprecated

Macro generated for the function crate::ops::logsumexp_axes. See the function documentation for more details.

logsumexp_axisDeprecated

Macro generated for the function crate::ops::logsumexp_axis. See the function documentation for more details.

ltDeprecated

Macro generated for the function crate::ops::lt. See the function documentation for more details.

matmulDeprecated

Macro generated for the function crate::ops::matmul. See the function documentation for more details.

maxDeprecated

Macro generated for the function crate::ops::max. See the function documentation for more details.

max_axesDeprecated

Macro generated for the function crate::ops::max_axes. See the function documentation for more details.

max_axisDeprecated

Macro generated for the function crate::ops::max_axis. See the function documentation for more details.

maximumDeprecated

Macro generated for the function crate::ops::maximum. See the function documentation for more details.

meanDeprecated

Macro generated for the function crate::ops::mean. See the function documentation for more details.

mean_axesDeprecated

Macro generated for the function crate::ops::mean_axes. See the function documentation for more details.

mean_axisDeprecated

Macro generated for the function crate::ops::mean_axis. See the function documentation for more details.

medianDeprecated

Macro generated for the function crate::ops::median. See the function documentation for more details.

median_axesDeprecated

Macro generated for the function crate::ops::median_axes. See the function documentation for more details.

median_axisDeprecated

Macro generated for the function crate::ops::median_axis. See the function documentation for more details.

minDeprecated

Macro generated for the function crate::ops::min. See the function documentation for more details.

min_axesDeprecated

Macro generated for the function crate::ops::min_axes. See the function documentation for more details.

min_axisDeprecated

Macro generated for the function crate::ops::min_axis. See the function documentation for more details.

minimumDeprecated

Macro generated for the function crate::ops::minimum. See the function documentation for more details.

move_axisDeprecated

Macro generated for the function crate::ops::move_axis. See the function documentation for more details.

multiplyDeprecated

Macro generated for the function crate::ops::multiply. See the function documentation for more details.

neDeprecated

Macro generated for the function crate::ops::ne. See the function documentation for more details.

negativeDeprecated

Macro generated for the function crate::ops::negative. See the function documentation for more details.

onesDeprecated

Macro generated for the function crate::ops::ones. See the function documentation for more details.

ones_dtypeDeprecated

Macro generated for the function crate::ops::ones_dtype. See the function documentation for more details.

ones_likeDeprecated

Macro generated for the function crate::ops::ones_like. See the function documentation for more details.

outerDeprecated

Macro generated for the function crate::ops::outer. See the function documentation for more details.

padDeprecated

Macro generated for the function crate::ops::pad. See the function documentation for more details.

partitionDeprecated

Macro generated for the function crate::ops::partition. See the function documentation for more details.

partition_axisDeprecated

Macro generated for the function crate::ops::partition_axis. See the function documentation for more details.

powerDeprecated

Macro generated for the function crate::ops::power. See the function documentation for more details.

prodDeprecated

Macro generated for the function crate::ops::prod. See the function documentation for more details.

prod_axesDeprecated

Macro generated for the function crate::ops::prod_axes. See the function documentation for more details.

prod_axisDeprecated

Macro generated for the function crate::ops::prod_axis. See the function documentation for more details.

quantizeDeprecated

Macro generated for the function crate::ops::quantize. See the function documentation for more details.

quantized_matmulDeprecated

Macro generated for the function crate::ops::quantized_matmul. See the function documentation for more details.

radiansDeprecated

Macro generated for the function crate::ops::radians. See the function documentation for more details.

realDeprecated

Macro generated for the function crate::ops::real. See the function documentation for more details.

reciprocalDeprecated

Macro generated for the function crate::ops::reciprocal. See the function documentation for more details.

remainderDeprecated

Macro generated for the function crate::ops::remainder. See the function documentation for more details.

repeatDeprecated

Macro generated for the function crate::ops::repeat. See the function documentation for more details.

repeat_axisDeprecated

Macro generated for the function crate::ops::repeat_axis. See the function documentation for more details.

reshapeDeprecated

Macro generated for the function crate::ops::reshape. See the function documentation for more details.

roundDeprecated

Macro generated for the function crate::ops::round. See the function documentation for more details.

rsqrtDeprecated

Macro generated for the function crate::ops::rsqrt. See the function documentation for more details.

segmented_mmDeprecated

Macro generated for the function crate::ops::segmented_mm. See the function documentation for more details.

sigmoidDeprecated

Macro generated for the function crate::ops::sigmoid. See the function documentation for more details.

signDeprecated

Macro generated for the function crate::ops::sign. See the function documentation for more details.

sinDeprecated

Macro generated for the function crate::ops::sin. See the function documentation for more details.

sinhDeprecated

Macro generated for the function crate::ops::sinh. See the function documentation for more details.

softmaxDeprecated

Macro generated for the function crate::ops::softmax. See the function documentation for more details.

softmax_axesDeprecated

Macro generated for the function crate::ops::softmax_axes. See the function documentation for more details.

softmax_axisDeprecated

Macro generated for the function crate::ops::softmax_axis. See the function documentation for more details.

sortDeprecated

Macro generated for the function crate::ops::sort. See the function documentation for more details.

sort_axisDeprecated

Macro generated for the function crate::ops::sort_axis. See the function documentation for more details.

splitDeprecated

Macro generated for the function crate::ops::split. See the function documentation for more details.

split_sectionsDeprecated

Macro generated for the function crate::ops::split_sections. See the function documentation for more details.

sqrtDeprecated

Macro generated for the function crate::ops::sqrt. See the function documentation for more details.

squareDeprecated

Macro generated for the function crate::ops::square. See the function documentation for more details.

squeezeDeprecated

Macro generated for the function crate::ops::squeeze. See the function documentation for more details.

squeeze_axesDeprecated

Macro generated for the function crate::ops::squeeze_axes. See the function documentation for more details.

stackDeprecated

Macro generated for the function crate::ops::stack. See the function documentation for more details.

stack_axisDeprecated

Macro generated for the function crate::ops::stack_axis. See the function documentation for more details.

stdDeprecated

Macro generated for the function crate::ops::std. See the function documentation for more details.

std_axesDeprecated

Macro generated for the function crate::ops::std_axes. See the function documentation for more details.

std_axisDeprecated

Macro generated for the function crate::ops::std_axis. See the function documentation for more details.

subtractDeprecated

Macro generated for the function crate::ops::subtract. See the function documentation for more details.

sumDeprecated

Macro generated for the function crate::ops::sum. See the function documentation for more details.

sum_axesDeprecated

Macro generated for the function crate::ops::sum_axes. See the function documentation for more details.

sum_axisDeprecated

Macro generated for the function crate::ops::sum_axis. See the function documentation for more details.

swap_axesDeprecated

Macro generated for the function crate::ops::swap_axes. See the function documentation for more details.

tanDeprecated

Macro generated for the function crate::ops::tan. See the function documentation for more details.

tanhDeprecated

Macro generated for the function crate::ops::tanh. See the function documentation for more details.

tensordot_axesDeprecated

Macro generated for the function crate::ops::tensordot_axes. See the function documentation for more details.

tensordot_axisDeprecated

Macro generated for the function crate::ops::tensordot_axis. See the function documentation for more details.

tileDeprecated

Macro generated for the function crate::ops::tile. See the function documentation for more details.

to_fp8Deprecated

Macro generated for the function crate::ops::to_fp8. See the function documentation for more details.

transposeDeprecated

Macro generated for the function crate::ops::transpose. See the function documentation for more details.

transpose_axesDeprecated

Macro generated for the function crate::ops::transpose_axes. See the function documentation for more details.

triDeprecated

Macro generated for the function crate::ops::tri. See the function documentation for more details.

trilDeprecated

Macro generated for the function crate::ops::tril. See the function documentation for more details.

triuDeprecated

Macro generated for the function crate::ops::triu. See the function documentation for more details.

unflattenDeprecated

Macro generated for the function crate::ops::unflatten. See the function documentation for more details.

varDeprecated

Macro generated for the function crate::ops::var. See the function documentation for more details.

var_axesDeprecated

Macro generated for the function crate::ops::var_axes. See the function documentation for more details.

var_axisDeprecated

Macro generated for the function crate::ops::var_axis. See the function documentation for more details.

whichDeprecated

Macro generated for the function crate::ops::which. See the function documentation for more details.

zerosDeprecated

Macro generated for the function crate::ops::zeros. See the function documentation for more details.

zeros_dtypeDeprecated

Macro generated for the function crate::ops::zeros_dtype. See the function documentation for more details.

zeros_likeDeprecated

Macro generated for the function crate::ops::zeros_like. See the function documentation for more details.

Structs

ContiguousOptions

Options controlling contiguous layout materialization.

CountNonzeroOptions

Axis selection and dimension retention for Array::count_nonzero.

LinspaceOptions

Sample count and endpoint behavior for linspace.

LogCumsumExpOptions

Axis and scan direction for Array::logcumsumexp.

TraceOptions

Diagonal selection and dtype control for Array::trace.

Enums

PadMode

The padding mode.

PadWidth

Number of padding values to add to the edges of each axis.

SearchSide

Insertion side for Array::search_sorted.

Traits

ClipBound

A custom trait for the bound of the clip operation.

Functions

abs

Element-wise absolute value.

abs_deviceDeprecated

Compatibility shim for abs.

acos

Element-wise inverse cosine.

acos_deviceDeprecated

Compatibility shim for acos.

acosh

Element-wise inverse hyperbolic cosine.

acosh_deviceDeprecated

Compatibility shim for acosh.

add

See Array::add.

add_deviceDeprecated

Compatibility shim for add.

addmm

Matrix multiplication with addition and optional scaling.

addmm_deviceDeprecated

Compatibility shim for addmm.

all

See Array::all

all_axes

See Array::all_axes

all_axes_deviceDeprecated

Compatibility shim for all_axes.

all_axis

See Array::all_axis

all_axis_deviceDeprecated

Compatibility shim for all_axis.

all_close

See Array::all_close

all_close_deviceDeprecated

Compatibility shim for all_close.

all_deviceDeprecated

Compatibility shim for all.

any

See Array::any

any_axes

See Array::any

any_axes_deviceDeprecated

Compatibility shim for any_axes.

any_axis

See Array::any

any_axis_deviceDeprecated

Compatibility shim for any_axis.

any_deviceDeprecated

Compatibility shim for any.

arange

See Array::arange

arange_deviceDeprecated

Compatibility shim for arange.

argpartition

Returns the indices that partition the flattened array. Returns an error if the arguments are invalid.

argpartition_axis

Returns the indices that partition the array. Returns an error if the arguments are invalid.

argpartition_axis_deviceDeprecated

Compatibility shim for argpartition_axis.

argpartition_deviceDeprecated

Compatibility shim for argpartition.

argsort

Returns the indices that sort the flattened array. Returns an error if the arguments are invalid.

argsort_axis

Returns the indices that sort the array. Returns an error if the arguments are invalid.

argsort_axis_deviceDeprecated

Compatibility shim for argsort_axis.

argsort_deviceDeprecated

Compatibility shim for argsort.

array_eq

See Array::array_eq

array_eq_deviceDeprecated

Compatibility shim for array_eq.

as_strided

Create a view into the array with the given shape and strides.

as_strided_deviceDeprecated

Compatibility shim for as_strided.

asin

Element-wise inverse sine.

asin_deviceDeprecated

Compatibility shim for asin.

asinh

Element-wise inverse hyperbolic sine.

asinh_deviceDeprecated

Compatibility shim for asinh.

at_least_1d

Convert array to have at least one dimension.

at_least_1d_deviceDeprecated

Compatibility shim for at_least_1d.

at_least_2d

Convert array to have at least two dimensions.

at_least_2d_deviceDeprecated

Compatibility shim for at_least_2d.

at_least_3d

Convert array to have at least three dimensions.

at_least_3d_deviceDeprecated

Compatibility shim for at_least_3d.

atan

Element-wise inverse tangent.

atan2

Element-wise inverse tangent of b/a choosing the quadrant correctly.

atan2_deviceDeprecated

Compatibility shim for atan2.

atan_deviceDeprecated

Compatibility shim for atan.

atanh

Element-wise inverse hyperbolic tangent.

atanh_deviceDeprecated

Compatibility shim for atanh.

block_masked_mm

Matrix multiplication with block masking.

block_masked_mm_deviceDeprecated

Compatibility shim for block_masked_mm.

broadcast_arrays

Broadcast a vector of arrays against one another. Returns an error if the shapes are broadcastable.

broadcast_arrays_deviceDeprecated

Compatibility shim for broadcast_arrays.

broadcast_to

Broadcast an array to the given shape. Returns an error if the shapes are not broadcastable.

broadcast_to_deviceDeprecated

Compatibility shim for broadcast_to.

ceil

Element-wise ceiling.

ceil_deviceDeprecated

Compatibility shim for ceil.

clip

Clip the values of the array between the given minimum and maximum.

clip_deviceDeprecated

Compatibility shim for clip.

concatenate

Concatenate the arrays along the given axis. Returns an error if the shapes are invalid.

concatenate_axisDeprecated

Compatibility alias for concatenate.

concatenate_axis_deviceDeprecated

Compatibility shim for concatenate.

concatenate_deviceDeprecated

Compatibility shim preserving the old flattening behavior of concatenate.

concatenate_flat

Flatten the arrays and concatenate them. Use concatenate to concatenate along an axis.

conv1d

1D convolution over an input with several channels returning an error if the inputs are invalid.

conv1d_deviceDeprecated

Compatibility shim for conv1d.

conv2d

2D convolution over an input with several channels returning an error if the inputs are invalid.

conv2d_deviceDeprecated

Compatibility shim for conv2d.

conv3d

3D convolution over an input with several channels.

conv3d_deviceDeprecated

Compatibility shim for conv3d.

conv_general

General convolution over an input with several channels returning an error if the inputs are invalid.

conv_general_deviceDeprecated

Compatibility shim for conv_general.

conv_transpose1d

1D transposed convolution over an input with several channels.

conv_transpose1d_deviceDeprecated

Compatibility shim for conv_transpose1d.

conv_transpose2d

2D transposed convolution over an input with several channels.

conv_transpose2d_deviceDeprecated

Compatibility shim for conv_transpose2d.

conv_transpose3d

3D transposed convolution over an input with several channels.

conv_transpose3d_deviceDeprecated

Compatibility shim for conv_transpose3d.

cos

Element-wise cosine.

cos_deviceDeprecated

Compatibility shim for cos.

cosh

Element-wise hyperbolic cosine.

cosh_deviceDeprecated

Compatibility shim for cosh.

cummax

See Array::cummax

cummax_deviceDeprecated

Compatibility shim for cummax.

cummin

See Array::cummin

cummin_deviceDeprecated

Compatibility shim for cummin.

cumprod

See Array::cumprod

cumprod_deviceDeprecated

Compatibility shim for cumprod.

cumsum

See Array::cumsum

cumsum_deviceDeprecated

Compatibility shim for cumsum.

degrees

Convert angles from radians to degrees.

degrees_deviceDeprecated

Compatibility shim for degrees.

dequantize

Dequantize the matrix w using the provided scales and biases and the group_size and bits configuration.

dequantize_deviceDeprecated

Compatibility shim for dequantize.

diag

See Array::diag

diag_deviceDeprecated

Compatibility shim for diag.

diagonal

See Array::diagonal

diagonal_deviceDeprecated

Compatibility shim for diagonal.

divide

See Array::divide.

divide_deviceDeprecated

Compatibility shim for divide.

divmod

Element-wise quotient and remainder.

divmod_deviceDeprecated

Compatibility shim for divmod.

einsum

Perform the Einstein summation convention on the operands.

einsum_deviceDeprecated

Compatibility shim for einsum.

eq

See Array::eq

eq_deviceDeprecated

Compatibility shim for eq.

erf

Element-wise error function.

erf_deviceDeprecated

Compatibility shim for erf.

erfinv

Element-wise inverse error function.

erfinv_deviceDeprecated

Compatibility shim for erfinv.

exp

See Array::exp.

exp_deviceDeprecated

Compatibility shim for exp.

expand_dims

Similar to expand_dims_axes, but only takes a single axis.

expand_dims_axes

Add a size one dimension at the given axis, returns an error if the axes are invalid.

expand_dims_axes_deviceDeprecated

Compatibility shim for expand_dims_axes.

expand_dims_deviceDeprecated

Compatibility shim for expand_dims.

expm1

Element-wise exponential minus 1.

expm1_deviceDeprecated

Compatibility shim for expm1.

eye

See Array::eye

eye_deviceDeprecated

Compatibility shim for eye.

flatten

Flatten an array. Returns an error if the axes are invalid.

flatten_deviceDeprecated

Compatibility shim for flatten.

floor

See Array::floor.

floor_deviceDeprecated

Compatibility shim for floor.

floor_divide

See Array::floor_divide.

floor_divide_deviceDeprecated

Compatibility shim for floor_divide.

from_fp8

Convert an FP8 (E4M3) encoded array back to a floating point type.

from_fp8_deviceDeprecated

Compatibility shim for from_fp8.

full

See Array::full

full_deviceDeprecated

Compatibility shim for full.

full_like

An array filled with the given value, with the same shape as the input.

full_like_deviceDeprecated

Compatibility shim for full_like.

gather_mm

Matrix multiplication with gathered indices.

gather_mm_deviceDeprecated

Compatibility shim for gather_mm.

gather_qmm

Perform quantized matrix multiplication with gathered indices.

gather_qmm_deviceDeprecated

Compatibility shim for gather_qmm.

ge

See Array::ge

ge_deviceDeprecated

Compatibility shim for ge.

gt

See Array::gt

gt_deviceDeprecated

Compatibility shim for gt.

identity

See Array::identity

identity_deviceDeprecated

Compatibility shim for identity.

imag

Element-wise imaginary part from a complex array.

imag_deviceDeprecated

Compatibility shim for imag.

inner

Ordinary inner product of vectors for 1-D arrays, in higher dimensions a sum product over the last axes.

inner_deviceDeprecated

Compatibility shim for inner.

is_close

See Array::is_close

is_close_deviceDeprecated

Compatibility shim for is_close.

is_inf

Return a boolean array indicating which elements are +/- inifnity.

is_inf_deviceDeprecated

Compatibility shim for is_inf.

is_nan

Return a boolean array indicating which elements are NaN.

is_nan_deviceDeprecated

Compatibility shim for is_nan.

is_neg_inf

Return a boolean array indicating which elements are negative infinity.

is_neg_inf_deviceDeprecated

Compatibility shim for is_neg_inf.

is_pos_inf

Return a boolean array indicating which elements are positive infinity.

is_pos_inf_deviceDeprecated

Compatibility shim for is_pos_inf.

kron

Perform the Kronecker product of two arrays.

kron_deviceDeprecated

Compatibility shim for kron.

le

See Array::le

le_deviceDeprecated

Compatibility shim for le.

linspace

Generate evenly spaced values between start and stop.

linspace_deviceDeprecated

Compatibility shim for linspace.

log

See Array::log.

log2

See Array::log2.

log1p

See Array::log1p.

log1p_deviceDeprecated

Compatibility shim for log1p.

log2_deviceDeprecated

Compatibility shim for log2.

log10

See Array::log10.

log10_deviceDeprecated

Compatibility shim for log10.

log_deviceDeprecated

Compatibility shim for log.

logaddexp

Element-wise log-add-exp.

logaddexp_deviceDeprecated

Compatibility shim for logaddexp.

logical_and

See Array::logical_and

logical_and_deviceDeprecated

Compatibility shim for logical_and.

logical_not

See Array::logical_not

logical_not_deviceDeprecated

Compatibility shim for logical_not.

logical_or

See Array::logical_or

logical_or_deviceDeprecated

Compatibility shim for logical_or.

logical_xor

Cast both operands to boolean, broadcast them, and compare for inequality.

logsumexp

See Array::logsumexp

logsumexp_axes

See Array::logsumexp_axes

logsumexp_axes_deviceDeprecated

Compatibility shim for logsumexp_axes.

logsumexp_axis

See Array::logsumexp_axis

logsumexp_axis_deviceDeprecated

Compatibility shim for logsumexp_axis.

logsumexp_deviceDeprecated

Compatibility shim for logsumexp.

lt

See Array::lt

lt_deviceDeprecated

Compatibility shim for lt.

matmul

See Array::matmul.

matmul_deviceDeprecated

Compatibility shim for matmul.

max

See Array::max

max_axes

See Array::max_axes

max_axes_deviceDeprecated

Compatibility shim for max_axes.

max_axis

See Array::max_axis

max_axis_deviceDeprecated

Compatibility shim for max_axis.

max_deviceDeprecated

Compatibility shim for max.

maximum

Element-wise maximum.

maximum_deviceDeprecated

Compatibility shim for maximum.

mean

See Array::mean

mean_axes

See Array::mean_axes

mean_axes_deviceDeprecated

Compatibility shim for mean_axes.

mean_axis

See Array::mean_axis

mean_axis_deviceDeprecated

Compatibility shim for mean_axis.

mean_deviceDeprecated

Compatibility shim for mean.

median

See Array::median

median_axes

See Array::median_axes

median_axes_deviceDeprecated

Compatibility shim for median_axes.

median_axis

See Array::median_axis

median_axis_deviceDeprecated

Compatibility shim for median_axis.

median_deviceDeprecated

Compatibility shim for median.

min

See Array::min

min_axes

See Array::min

min_axes_deviceDeprecated

Compatibility shim for min_axes.

min_axis

See Array::min_axis

min_axis_deviceDeprecated

Compatibility shim for min_axis.

min_deviceDeprecated

Compatibility shim for min.

minimum

Element-wise minimum.

minimum_deviceDeprecated

Compatibility shim for minimum.

move_axis

Move an axis to a new position. Returns an error if the axes are invalid.

move_axis_deviceDeprecated

Compatibility shim for move_axis.

multiply

See Array::multiply.

multiply_deviceDeprecated

Compatibility shim for multiply.

ne

See Array::ne

ne_deviceDeprecated

Compatibility shim for ne.

negative

See Array::negative.

negative_deviceDeprecated

Compatibility shim for negative.

ones

See Array::ones

ones_deviceDeprecated

Compatibility shim for ones.

ones_dtype

Similar to Array::ones but with a specified dtype.

ones_dtype_deviceDeprecated

Compatibility shim for ones_dtype.

ones_like

An array of ones like the input.

ones_like_deviceDeprecated

Compatibility shim for ones_like.

outer

Compute the outer product of two 1-D arrays, if the array’s passed are not 1-D a flatten op will be run beforehand.

outer_deviceDeprecated

Compatibility shim for outer.

pad

Pad an array with a constant value. Returns an error if the width is invalid.

pad_deviceDeprecated

Compatibility shim for pad.

partition

Returns a partitioned copy of the flattened array such that the smaller kth elements are first. Returns an error if the arguments are invalid.

partition_axis

Returns a partitioned copy of the array such that the smaller kth elements are first. Returns an error if the arguments are invalid.

partition_axis_deviceDeprecated

Compatibility shim for partition_axis.

partition_deviceDeprecated

Compatibility shim for partition.

power

See Array::power.

power_deviceDeprecated

Compatibility shim for power.

prod

See Array::prod

prod_axes

See Array::prod_axes

prod_axes_deviceDeprecated

Compatibility shim for prod_axes.

prod_axis

See Array::prod_axis

prod_axis_deviceDeprecated

Compatibility shim for prod_axis.

prod_deviceDeprecated

Compatibility shim for prod.

quantize

Quantize the matrix w using bits bits per element.

quantize_deviceDeprecated

Compatibility shim for quantize.

quantized_matmul

Perform the matrix multiplication with the quantized matrix w. The quantization uses one floating point scale and bias per group_size of elements. Each element in w takes bits bits and is packed in an unsigned 32 bit integer.

quantized_matmul_deviceDeprecated

Compatibility shim for quantized_matmul.

radians

Convert angles from degrees to radians.

radians_deviceDeprecated

Compatibility shim for radians.

real

Element-wise real part from a complex array.

real_deviceDeprecated

Compatibility shim for real.

reciprocal

See Array::reciprocal.

reciprocal_deviceDeprecated

Compatibility shim for reciprocal.

remainder

See Array::remainder.

remainder_deviceDeprecated

Compatibility shim for remainder.

repeat

See Array::repeat

repeat_axis

See Array::repeat

repeat_axis_deviceDeprecated

Compatibility shim for repeat_axis.

repeat_deviceDeprecated

Compatibility shim for repeat.

reshape

Reshape an array while preserving the size. Returns an error if the new shape is invalid.

reshape_deviceDeprecated

Compatibility shim for reshape.

round

See Array::round.

round_deviceDeprecated

Compatibility shim for round.

rsqrt

See Array::rsqrt.

rsqrt_deviceDeprecated

Compatibility shim for rsqrt.

segmented_mm

Perform a segmented matrix multiplication.

segmented_mm_deviceDeprecated

Compatibility shim for segmented_mm.

select

Select from a or b according to condition returning an error if the arrays are not broadcastable.

sigmoid

Element-wise logistic sigmoid.

sigmoid_deviceDeprecated

Compatibility shim for sigmoid.

sign

Element-wise sign.

sign_deviceDeprecated

Compatibility shim for sign.

sin

See Array::sin.

sin_deviceDeprecated

Compatibility shim for sin.

sinh

Element-wise hyperbolic sine.

sinh_deviceDeprecated

Compatibility shim for sinh.

softmax

Similar to softmax_axes but with no axis specified.

softmax_axes

Perform the softmax along the given axis.

softmax_axes_deviceDeprecated

Compatibility shim for softmax_axes.

softmax_axis

Similar to softmax_axes but with a single axis.

softmax_axis_deviceDeprecated

Compatibility shim for softmax_axis.

softmax_deviceDeprecated

Compatibility shim for softmax.

sort

Returns a sorted copy of the flattened array. Returns an error if the arguments are invalid.

sort_axis

Returns a sorted copy of the array. Returns an error if the arguments are invalid.

sort_axis_deviceDeprecated

Compatibility shim for sort_axis.

sort_deviceDeprecated

Compatibility shim for sort.

splitDeprecated

Compatibility alias for split_equal.

split_at_indices

Split an array along a given axis. Returns an error if the indices are invalid.

split_deviceDeprecated

Compatibility shim for split_equal.

split_equal

Split an array into equal parts along a given axis. Returns an error if the array cannot be split into equal parts.

split_sectionsDeprecated

Compatibility alias for split_at_indices.

split_sections_deviceDeprecated

Compatibility shim for split_at_indices.

sqrt

See Array::sqrt.

sqrt_deviceDeprecated

Compatibility shim for sqrt.

square

See Array::square.

square_deviceDeprecated

Compatibility shim for square.

squeeze

Similar to squeeze_axes, but removes all length one axes.

squeeze_axes

Remove length one axes from an array. Returns an error if the axes are invalid.

squeeze_axes_deviceDeprecated

Compatibility shim for squeeze_axes.

squeeze_deviceDeprecated

Compatibility shim for squeeze.

stack

Stacks the arrays along a new axis. Returns an error if the arguments are invalid.

stack_axisDeprecated

Compatibility alias for stack.

stack_axis_deviceDeprecated

Compatibility shim for stack.

stack_deviceDeprecated

Compatibility shim preserving the old axis-zero behavior of stack.

std

Similar to std_axes but reduces over all axes.

std_axes

Compute the standard deviation(s) over the given axes.

std_axes_deviceDeprecated

Compatibility shim for std_axes.

std_axis

Similar to std_axes but only reduces over a single axis.

std_axis_deviceDeprecated

Compatibility shim for std_axis.

std_deviceDeprecated

Compatibility shim for [std].

subtract

See Array::subtract.

subtract_deviceDeprecated

Compatibility shim for subtract.

sum

See Array::sum

sum_axes

See Array::sum_axes

sum_axes_deviceDeprecated

Compatibility shim for sum_axes.

sum_axis

See Array::sum_axis

sum_axis_deviceDeprecated

Compatibility shim for sum_axis.

sum_deviceDeprecated

Compatibility shim for sum.

swap_axes

Swap two axes of an array. Returns an error if the axes are invalid.

swap_axes_deviceDeprecated

Compatibility shim for swap_axes.

tan

See [Array::tan].

tan_deviceDeprecated

Compatibility shim for tan.

tanh

Element-wise hyperbolic tangent.

tanh_deviceDeprecated

Compatibility shim for tanh.

tensordot_axes

Compute the tensor dot product along the specified axes.

tensordot_axes_deviceDeprecated

Compatibility shim for tensordot_axes.

tensordot_axis

Similar to tensordot_axes but with a single axis.

tensordot_axis_deviceDeprecated

Compatibility shim for tensordot_axis.

tile

Construct an array by repeating a the number of times given by reps.

tile_deviceDeprecated

Compatibility shim for tile.

to_fp8

Convert an array to FP8 (E4M3) format.

to_fp8_deviceDeprecated

Compatibility shim for to_fp8.

transpose

Transpose with all axes reversed

transpose_axes

Transpose the dimensions of the array. Returns an error if the axes are invalid.

transpose_axes_deviceDeprecated

Compatibility shim for transpose_axes.

transpose_deviceDeprecated

Compatibility shim for transpose.

tri

See Array::tri

tri_deviceDeprecated

Compatibility shim for tri.

tril

Zeros the array above the given diagonal

tril_deviceDeprecated

Compatibility shim for tril.

triu

Zeros the array below the given diagonal

triu_deviceDeprecated

Compatibility shim for triu.

unflatten

Unflatten an axis of an array to a shape.

unflatten_deviceDeprecated

Compatibility shim for unflatten.

var

See Array::var

var_axes

See Array::var_axes

var_axes_deviceDeprecated

Compatibility shim for var_axes.

var_axis

See Array::var_axis

var_axis_deviceDeprecated

Compatibility shim for var_axis.

var_deviceDeprecated

Compatibility shim for var.

vecdot

Compute a vector dot product along axis.

whereDeprecated

Compatibility alias for select.

where_deviceDeprecated

Compatibility shim for select.

whichDeprecated

Compatibility alias for select.

which_deviceDeprecated

Compatibility shim for select.

zeros

See Array::zeros

zeros_deviceDeprecated

Compatibility shim for zeros.

zeros_dtype

Similar to Array::zeros but with a specified dtype.

zeros_dtype_deviceDeprecated

Compatibility shim for zeros_dtype.

zeros_like

An array of zeros like the input.

zeros_like_deviceDeprecated

Compatibility shim for zeros_like.