Module ops

Operations

Modules

indexing

Indexing Arrays

Macros

abs

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

acos

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

acosh

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

add

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

addmm

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

all

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

all_axes

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

all_axis

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

all_close

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

any

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

any_axes

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

any_axis

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

arange

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

argpartition

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

argpartition_axis

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

argsort

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

argsort_axis

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

array_eq

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

as_strided

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

asin

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

asinh

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

at_least_1d

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

at_least_2d

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

at_least_3d

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

atan

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

atanh

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

block_masked_mm

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

broadcast_arrays

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

broadcast_to

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

ceil

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

clip

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

concatenate

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

concatenate_axis

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

conv1d

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

conv2d

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

conv3d

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

conv_general

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

conv_transpose1d

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

conv_transpose2d

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

conv_transpose3d

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

cos

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

cosh

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

cummax

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

cummin

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

cumprod

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

cumsum

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

degrees

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

dequantize

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

diag

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

diagonal

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

divide

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

divmod

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

einsum

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

eq

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

erf

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

erfinv

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

exp

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

expand_dims

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

expand_dims_axes

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

expm1

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

eye

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

flatten

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

floor

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

floor_divide

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

full

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

ge

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

gt

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

identity

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

imag

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

inner

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

is_close

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

is_inf

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

is_nan

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

is_neg_inf

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

is_pos_inf

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

kron

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

le

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

linspace

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

log

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

log2

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

log1p

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

log10

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

logaddexp

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

logical_and

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

logical_not

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

logical_or

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

logsumexp

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

logsumexp_axes

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

logsumexp_axis

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

lt

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

matmul

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

max

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

max_axes

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

max_axis

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

maximum

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

mean

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

mean_axes

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

mean_axis

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

min

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

min_axes

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

min_axis

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

minimum

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

move_axis

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

multiply

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

ne

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

negative

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

ones

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

ones_dtype

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

ones_like

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

outer

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

pad

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

partition

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

partition_axis

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

power

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

prod

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

prod_axes

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

prod_axis

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

quantize

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

quantized_matmul

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

radians

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

real

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

reciprocal

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

remainder

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

repeat

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

repeat_axis

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

reshape

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

round

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

rsqrt

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

sigmoid

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

sign

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

sin

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

sinh

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

softmax

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

softmax_axes

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

softmax_axis

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

sort

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

sort_axis

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

split

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

split_sections

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

sqrt

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

square

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

squeeze

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

squeeze_axes

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

stack

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

stack_axis

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

std

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

std_axes

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

std_axis

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

subtract

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

sum

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

sum_axes

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

sum_axis

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

swap_axes

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

tan

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

tanh

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

tensordot_axes

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

tensordot_axis

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

tile

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

transpose

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

transpose_axes

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

tri

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

tril

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

triu

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

unflatten

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

var

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

var_axes

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

var_axis

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

which

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

zeros

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

zeros_dtype

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

zeros_like

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

Enums

PadMode

The padding mode.

PadWidth

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

Traits

ClipBound

A custom trait for the bound of the clip operation.

Functions

abs

Element-wise absolute value.

abs_device

Element-wise absolute value.

acos

Element-wise inverse cosine.

acos_device

Element-wise inverse cosine.

acosh

Element-wise inverse hyperbolic cosine.

acosh_device

Element-wise inverse hyperbolic cosine.

add

See Array::add.

add_device

See Array::add.

addmm

Matrix multiplication with addition and optional scaling.

addmm_device

Matrix multiplication with addition and optional scaling.

all

See Array::all

all_axes

See Array::all_axes

all_axes_device

See Array::all_axes

all_axis

See Array::all_axis

all_axis_device

See Array::all_axis

all_close

See Array::all_close

all_close_device

See Array::all_close

all_device

See Array::all

any

See Array::any

any_axes

See Array::any

any_axes_device

See Array::any

any_axis

See Array::any

any_axis_device

See Array::any

any_device

See Array::any

arange

See Array::arange

arange_device

See Array::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_device

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

argpartition_device

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

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_device

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

argsort_device

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

array_eq

See Array::array_eq

array_eq_device

See Array::array_eq

as_strided

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

as_strided_device

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

asin

Element-wise inverse sine.

asin_device

Element-wise inverse sine.

asinh

Element-wise inverse hyperbolic sine.

asinh_device

Element-wise inverse hyperbolic sine.

at_least_1d

Convert array to have at least one dimension.

at_least_1d_device

Convert array to have at least one dimension.

at_least_2d

Convert array to have at least two dimensions.

at_least_2d_device

Convert array to have at least two dimensions.

at_least_3d

Convert array to have at least three dimensions.

at_least_3d_device

Convert array to have at least three dimensions.

atan

Element-wise inverse tangent.

atan_device

Element-wise inverse tangent.

atanh

Element-wise inverse hyperbolic tangent.

atanh_device

Element-wise inverse hyperbolic tangent.

block_masked_mm

Matrix multiplication with block masking.

block_masked_mm_device

Matrix multiplication with block masking.

broadcast_arrays

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

broadcast_arrays_device

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

broadcast_to

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

broadcast_to_device

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

ceil

Element-wise ceiling.

ceil_device

Element-wise ceiling.

clip

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

clip_device

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

concatenate

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

concatenate_axis

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

concatenate_axis_device

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

concatenate_device

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

conv1d

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

conv1d_device

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

conv2d

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

conv2d_device

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

conv3d

3D convolution over an input with several channels.

conv3d_device

3D convolution over an input with several channels.

conv_general

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

conv_general_device

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

conv_transpose1d

1D transposed convolution over an input with several channels.

conv_transpose1d_device

1D transposed convolution over an input with several channels.

conv_transpose2d

2D transposed convolution over an input with several channels.

conv_transpose2d_device

2D transposed convolution over an input with several channels.

conv_transpose3d

3D transposed convolution over an input with several channels.

conv_transpose3d_device

3D transposed convolution over an input with several channels.

cos

Element-wise cosine.

cos_device

Element-wise cosine.

cosh

Element-wise hyperbolic cosine.

cosh_device

Element-wise hyperbolic cosine.

cummax

See Array::cummax

cummax_device

See Array::cummax

cummin

See Array::cummin

cummin_device

See Array::cummin

cumprod

See Array::cumprod

cumprod_device

See Array::cumprod

cumsum

See Array::cumsum

cumsum_device

See Array::cumsum

degrees

Convert angles from radians to degrees.

degrees_device

Convert angles from radians to degrees.

dequantize

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

dequantize_device

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

diag

See Array::diag

diag_device

See Array::diag

diagonal

See Array::diagonal

diagonal_device

See Array::diagonal

divide

See Array::divide.

divide_device

See Array::divide.

divmod

Element-wise quotient and remainder.

divmod_device

Element-wise quotient and remainder.

einsum

Perform the Einstein summation convention on the operands.

einsum_device

Perform the Einstein summation convention on the operands.

eq

See Array::eq

eq_device

See Array::eq

erf

Element-wise error function.

erf_device

Element-wise error function.

erfinv

Element-wise inverse error function.

erfinv_device

Element-wise inverse error function.

exp

See Array::exp.

exp_device

See Array::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_device

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

expand_dims_device

Similar to expand_dims_axes, but only takes a single axis.

expm1

Element-wise exponential minus 1.

expm1_device

Element-wise exponential minus 1.

eye

See Array::eye

eye_device

See Array::eye

flatten

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

flatten_device

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

floor

See Array::floor.

floor_device

See Array::floor.

floor_divide

See Array::floor_divide.

floor_divide_device

See Array::floor_divide.

full

See Array::full

full_device

See Array::full

ge

See Array::ge

ge_device

See Array::ge

gt

See Array::gt

gt_device

See Array::gt

identity

See Array::identity

identity_device

See Array::identity

imag

Element-wise imaginary part from a complex array.

imag_device

Element-wise imaginary part from a complex array.

inner

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

inner_device

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

is_close

See Array::is_close

is_close_device

See Array::is_close

is_inf

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

is_inf_device

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

is_nan

Return a boolean array indicating which elements are NaN.

is_nan_device

Return a boolean array indicating which elements are NaN.

is_neg_inf

Return a boolean array indicating which elements are negative infinity.

is_neg_inf_device

Return a boolean array indicating which elements are negative infinity.

is_pos_inf

Return a boolean array indicating which elements are positive infinity.

is_pos_inf_device

Return a boolean array indicating which elements are positive infinity.

kron

Perform the Kronecker product of two arrays.

kron_device

Perform the Kronecker product of two arrays.

le

See Array::le

le_device

See Array::le

linspace

See Array::linspace

linspace_device

See Array::linspace

log

See Array::log.

log2

See Array::log2.

log1p

See Array::log1p.

log1p_device

See Array::log1p.

log2_device

See Array::log2.

log10

See Array::log10.

log10_device

See Array::log10.

log_device

See Array::log.

logaddexp

Element-wise log-add-exp.

logaddexp_device

Element-wise log-add-exp.

logical_and

See Array::logical_and

logical_and_device

See Array::logical_and

logical_not

See Array::logical_not

logical_not_device

See Array::logical_not

logical_or

See Array::logical_or

logical_or_device

See Array::logical_or

logsumexp

See Array::logsumexp

logsumexp_axes

See Array::logsumexp_axes

logsumexp_axes_device

See Array::logsumexp_axes

logsumexp_axis

See Array::logsumexp_axis

logsumexp_axis_device

See Array::logsumexp_axis

logsumexp_device

See Array::logsumexp

lt

See Array::lt

lt_device

See Array::lt

matmul

See Array::matmul.

matmul_device

See Array::matmul.

max

See Array::max

max_axes

See Array::max_axes

max_axes_device

See Array::max_axes

max_axis

See Array::max_axis

max_axis_device

See Array::max_axis

max_device

See Array::max

maximum

Element-wise maximum.

maximum_device

Element-wise maximum.

mean

See Array::mean

mean_axes

See Array::mean_axes

mean_axes_device

See Array::mean_axes

mean_axis

See Array::mean_axis

mean_axis_device

See Array::mean_axis

mean_device

See Array::mean

min

See Array::min

min_axes

See Array::min

min_axes_device

See Array::min

min_axis

See Array::min_axis

min_axis_device

See Array::min_axis

min_device

See Array::min

minimum

Element-wise minimum.

minimum_device

Element-wise minimum.

move_axis

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

move_axis_device

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

multiply

See Array::multiply.

multiply_device

See Array::multiply.

ne

See Array::ne

ne_device

See Array::ne

negative

See Array::negative.

negative_device

See Array::negative.

ones

See Array::ones

ones_device

See Array::ones

ones_dtype

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

ones_dtype_device

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

ones_like

An array of ones like the input.

ones_like_device

An array of ones like the input.

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_device

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.

pad

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

pad_device

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

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_device

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

partition_device

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

power

See Array::power.

power_device

See Array::power.

prod

See Array::prod

prod_axes

See Array::prod_axes

prod_axes_device

See Array::prod_axes

prod_axis

See Array::prod_axis

prod_axis_device

See Array::prod_axis

prod_device

See Array::prod

quantize

Quantize the matrix w using bits bits per element.

quantize_device

Quantize the matrix w using bits bits per element.

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_device

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.

radians

Convert angles from degrees to radians.

radians_device

Convert angles from degrees to radians.

real

Element-wise real part from a complex array.

real_device

Element-wise real part from a complex array.

reciprocal

See Array::reciprocal.

reciprocal_device

See Array::reciprocal.

remainder

See Array::remainder.

remainder_device

See Array::remainder.

repeat

See Array::repeat

repeat_axis

See Array::repeat

repeat_axis_device

See Array::repeat

repeat_device

See Array::repeat

reshape

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

reshape_device

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

round

See Array::round.

round_device

See Array::round.

rsqrt

See Array::rsqrt.

rsqrt_device

See Array::rsqrt.

sigmoid

Element-wise logistic sigmoid.

sigmoid_device

Element-wise logistic sigmoid.

sign

Element-wise sign.

sign_device

Element-wise sign.

sin

See Array::sin.

sin_device

See Array::sin.

sinh

Element-wise hyperbolic sine.

sinh_device

Element-wise hyperbolic sine.

softmax

Similar to [softmax_axes] but with no axis specified.

softmax_axes

Perform the softmax along the given axis.

softmax_axes_device

Perform the softmax along the given axis.

softmax_axis

Similar to [softmax_axes] but with a single axis.

softmax_axis_device

Similar to [softmax_axes] but with a single axis.

softmax_device

Similar to [softmax_axes] but with no axis specified.

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_device

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

sort_device

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

split

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

split_device

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

split_sections

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

split_sections_device

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

sqrt

See Array::sqrt.

sqrt_device

See Array::sqrt.

square

See Array::square.

square_device

See Array::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_device

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

squeeze_device

Similar to squeeze_axes, but removes all length one axes.

stack

Stacks the arrays along a new axis. Returns an error if the arrays have different shapes.

stack_axis

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

stack_axis_device

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

stack_device

Stacks the arrays along a new axis. Returns an error if the arrays have different shapes.

std

Similar to [std_axes] but reduces over all axes.

std_axes

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

std_axes_device

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

std_axis

Similar to [std_axes] but only reduces over a single axis.

std_axis_device

Similar to [std_axes] but only reduces over a single axis.

std_device

Similar to [std_axes] but reduces over all axes.

subtract

See Array::subtract.

subtract_device

See Array::subtract.

sum

See Array::sum

sum_axes

See Array::sum_axes

sum_axes_device

See Array::sum_axes

sum_axis

See Array::sum_axis

sum_axis_device

See Array::sum_axis

sum_device

See Array::sum

swap_axes

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

swap_axes_device

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

tan

See [Array::tan].

tan_device

See [Array::tan].

tanh

Element-wise hyperbolic tangent.

tanh_device

Element-wise hyperbolic tangent.

tensordot_axes

Compute the tensor dot product along the specified axes.

tensordot_axes_device

Compute the tensor dot product along the specified axes.

tensordot_axis

Similar to [tensordot_axes] but with a single axis.

tensordot_axis_device

Similar to [tensordot_axes] but with a single axis.

tile

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

tile_device

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

transpose

Transpose with all axes reversed

transpose_axes

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

transpose_axes_device

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

transpose_device

Transpose with all axes reversed

tri

See Array::tri

tri_device

See Array::tri

tril

Zeros the array above the given diagonal

tril_device

Zeros the array above the given diagonal

triu

Zeros the array below the given diagonal

triu_device

Zeros the array below the given diagonal

unflatten

Unflatten an axis of an array to a shape.

unflatten_device

Unflatten an axis of an array to a shape.

var

See Array::var

var_axes

See Array::var_axes

var_axes_device

See Array::var_axes

var_axis

See Array::var_axis

var_axis_device

See Array::var_axis

var_device

See Array::var

where

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

where_device

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

which

Alias for [r#where]

which_device

Alias for [r#where]

zeros

See Array::zeros

zeros_device

See Array::zeros

zeros_dtype

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

zeros_dtype_device

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

zeros_like

An array of zeros like the input.

zeros_like_device

An array of zeros like the input.