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