在张量流中乘以稀疏矩阵

时间:2017-09-19 21:58:13

标签: tensorflow

我有一个非常大的稀疏二维矩阵(200k x 100k)(稀疏度大约为0.002),我需要与自身相乘(换位后)。

A = 200k x 100k
我需要计算:
A x A.transpose()

问题是,在Tensorflow中,稀疏矩阵乘法似乎没有稀疏矩阵 tf.sparse_tensor_dense_matmul()要求将一侧变成密集矩阵,这将达到内存限制 tf.sparse_matmul()似乎实际上需要密集的矩阵。

无法将整个矩阵(A)作为密集向量拟合到GPU内存中。

在Tensorflow中有没有办法使用其他稀疏矩阵的多个稀疏矩阵?我似乎无法找到一个。

0 个答案:

没有答案