torch.smm

torch.smm(input, mat) → Tensor

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

参数:

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

• 矩阵（张量）- 要进行矩阵乘法的密集矩阵