API ReferenceΒΆ

apply_where

Run one of two elementwise functions depending on a condition.

at

Update operations for read-only arrays.

atleast_nd

Recursively expand the dimension of an array to at least ndim.

broadcast_shapes

Compute the shape of the broadcasted arrays.

cov

Estimate a covariance matrix.

create_diagonal

Construct a diagonal array.

default_dtype

Return the default dtype for the given namespace and device.

expand_dims

Expand the shape of an array.

isclose

Return a boolean array where two arrays are element-wise equal within a tolerance.

kron

Kronecker product of two arrays.

nunique

Count the number of unique elements in an array.

one_hot

One-hot encode the given indices.

pad

Pad the input array.

setdiff1d

Find the set difference of two arrays.

sinc

Return the normalized sinc function.