Rate this Page

torch.smm#

torch.smm(input,mat)Tensor#

Performs a matrix multiplication of the sparse matrixinputwith the dense matrixmat.

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

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