Sparse Tensors in PyTorch

torch.mm and torch.spmm should work then