Rate this Page

torch.smm#

torch.smm(input, mat) Tensor#

Performs a matrix multiplication of the sparse matrix input with the dense matrix mat.

Parameters
  • input (Tensor) – a sparse matrix to be matrix multiplied

  • mat (Tensor) – a dense matrix to be matrix multiplied