-
Feng authored
a simple modification to the code of spmm() can make it support also 3D tensor (might think as batched): suppose dense matrix has dimensions (B, N, F), the nonzero has dimensions (B, num_edges), all graphs share the same topology (i.e., the same adjacency matrix)
9a4ea01b
