Skip to main content

module

linalg

Ops that perform linear algebra.

Functions

  • outer: Computes the outer product of two symbolic vectors.
  • matmul_broadcast: Computes the broadcasting of two symbolic tensors for a matmul.
  • matmul: Computes the matrix multiplication of two symbolic tensors.
  • batch_matmul: Computes the matrix multiplication of two symbolic tensors.
  • matmul_by_matrix: Computes the matrix multiplication of two symbolic tensors.
  • band_part: Masks out everything except a diagonal band of an input matrix.