tensorflow - 使用 adj 列表在 Tensorflow 中构造 adj 矩阵

Question

我的模型需要一个邻接矩阵，该矩阵目前在 numpy 中创建并作为占位符传递给 tensorflow。

随着问题规模的扩大，我认为内存和 VRAM 之间的 I/O 是一个瓶颈，因为复杂性是二次方的。例如，我使用 dim 400，这将产生 160.000 个矩阵值。

由于 adj 矩阵是稀疏的，我考虑过传递一个 adj 列表，然后在 GPU 上的 tf 中创建 adj 矩阵。

有什么建议么？谢谢

标签： tensorflow

Answer 1

TensorFlow 支持稀疏占位符。在此页面https://www.tensorflow.org/api_docs/python/tf/sparse_placeholder
中有一个示例显示如何使用tf.sparse_placeholder