标签: tensorflow sparse-matrix matrix-multiplication
我正在尝试使用tensorflow进行稀疏矩阵乘法。
但是,当我看到sparse_matmul_op.cc文件时,它似乎没有启动任何cuda内核,例如cuBlas或cuSparse。
是在CPU上计算的吗?不是GPU?