如何在TensorFlow中执行稀疏矩阵*稀疏矩阵乘法?

时间:2018-08-14 02:57:42

标签: python tensorflow matrix

目前,我正在尝试使用TensorFlow计算矩阵乘法,其中两个矩阵均为稀疏形式。我尝试使用tf.sparse_matmul(),但在那种情况下,我不得不转换回密集的形式,这破坏了我的项目目标。请建议我在这种情况下该怎么办。

1 个答案:

答案 0 :(得分:0)

您尝试过tf.sparse_matmul吗?

 sparse_matumul_result = tf.sparse_matmul(a, b)

如果还需要sparse_matmul_result作为稀疏对象,则可以执行以下操作:

 tf.contrib.layers.dense_to_sparse(sparse_matmul_result)