我建立了一个类似于 CIFAR10 的 GPU 实现的张量流模型。我有一个在每个 GPU 上执行的基本模型，而网络变量则在 CPU 上。只要我不使用稀疏张量作为层中的权重矩阵，一切都会正常工作。
我的稀疏权重矩阵是用函数构造的tf.sparse_to_dense() or tf.diag()。当我在 CPU 上运行它时，一切正常，但当我在 GPU 上运行它时，我收到消息：这些函数没有 GPU 实现.
还有其他选择吗？如何构建位于 CPU 上但在 GPU 上训练的稀疏权重矩阵？
None
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
