torch.smm

torch.smm(input, mat) → Tensor

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

引數

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

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