torch.smm

torch.smm(input, mat) → Tensor

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

参数

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

• mat (Tensor) – 要执行矩阵乘法的稠密矩阵