torch.smm

torch.smm(input, mat) → Tensor

对稀疏矩阵input和密集矩阵mat进行矩阵乘法运算。

参数

• 输入 (Tensor) – 待进行矩阵相乘的稀疏矩阵

• mat (张量) – 需要进行矩阵乘法的密集型矩阵