我 Build 了一个类似于CIFAR10的GPU实现的张量流模型 . 我有一个在每个GPU上执行的基本模型，而网络的变量在CPU上 . 只要我不使用稀疏张量作为图层中的权重矩阵，一切都可以正常工作 .
我的稀疏权重矩阵用函数 tf.sparse_to_dense() 或 tf.diag() 构造 . 当我在CPU上运行它时一切正常，但是当我在GPU上运行它时，我得到的消息是 no GPU implementation of these functions .
还有其他选择吗？如何构建位于CPU上但在GPU上训练的稀疏权重矩阵？