torch.smm

torch.smm(input, mat) → Tensor

執行稀疏矩陣 input 與稠密矩陣 mat 的矩陣乘法。

引數

• input (Tensor) – 要進行矩陣乘法的稀疏矩陣

• mat (Tensor) – 要進行矩陣乘法的稠密矩陣