Module linalg

Expand description

The linalg module.

Enums§

Norm
Specifies the type of norm to compute.

Constants§

DEFAULT_EPSILON
Default epsilon value to avoid division by zero

Functions§

cosine_similarity
Computes the cosine similarity between two tensors along a specified dimension.
diag
Returns the diag of a matrix.
l0_norm
Computes the L0 norm of a tensor along a specified dimension.
l1_norm
Computes the L1 norm of a tensor along a specified dimension.
l2_norm
Computes the L2 norm of a tensor along a specified dimension.
lp_norm
Computes the general L(p) norm of a tensor along a specified dimension.
lu_decomposition
Performs PLU decomposition of a square matrix.
max_abs_norm
Computes the L:INFINITY norm of a tensor along a specified dimension.
min_abs_norm
Computes the L:NEG_INFINITY norm of a tensor along a specified dimension.
outer
Computes the outer product (and batched outer product) for rank-1 or rank-2 tensor.
trace
Computes the trace of a matrix.
vector_norm
Computes the vector norm of a tensor along a specified dimension.
vector_normalize
Normalize a tensor versus its vector_norm.