torch.smm

torch.smm(input, mat) → Tensor

执行稀疏矩阵 input 与密集矩阵 mat 的矩阵乘法。

参数

• input (Tensor) – 要进行矩阵乘法的稀疏矩阵

• mat (Tensor) – 要进行矩阵乘法的密集矩阵