torch、torch_sparse、dgl三者spmm显存占用对比
简单比较一下torch原生，torch_sparse(pyG依赖模块)以及DGL中spmm的显存占用和时间torch原生，显存占用主要是对结点敏感（bakward的时候用的稠密张量）。# -*- coding: utf-8 -*-"""@Author: @Time: 2020/12/1@Description: """import torchimport numpy as npimport torch_sparseimport dgli
