Shortcuts

torch.smm

torch.smm(input, mat) 张量

对稀疏矩阵 input 与稠密矩阵 mat 进行矩阵乘法运算。

Parameters
  • 输入 (张量) – 一个需要进行矩阵乘法的稀疏矩阵

  • mat (张量) – 一个用于矩阵乘法的稠密矩阵

优云智算