Module linalg
Expand description
The linalg module.
Enums§
- Norm
- Specifies the type of norm to compute.
Functions§
- cosine_
similarity - Computes the cosine similarity between two tensors along a specified dimension.
- det
- Computes the determinant on the last two dimensions of the input tensor.
- 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
- Computes the LU decomposition of a square or rectangular matrix with partial pivoting.
- matvec
- Performs matrix-vector multiplication with optional batch dimensions.
- 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 for the last columns of 2 tensors.
- outer_
dim - Computes the outer product along a specific dimension, broadcasting over others.
- 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.