标签： tensorflow gpu sparse-matrix
我构建了一个类似于CIFAR10的GPU实现的张量流模型。我有一个在每个GPU上执行的基本模型，而网络的变量在CPU上。只要我不在图层中使用稀疏张量作为权重矩阵，一切都可以正常工作。
我的稀疏权重矩阵是使用函数 tf.sparse_to_dense（）或 tf.diag（）构建的。当我在CPU上运行时，一切正常，但是当我在GPU上运行它时，我得到的消息是没有这些功能的GPU实现。
还有其他选择吗？如何构建位于CPU上但在GPU上训练的稀疏权重矩阵？