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

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了如何在TensorFlow中执行稀疏矩阵*稀疏矩阵乘法?相关的知识,希望对你有一定的参考价值。

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

答案

你试过tf.sparse_matmul吗?

 sparse_matumul_result = tf.sparse_matmul(a, b)

如果你还需要sparse_matmul_result作为稀疏,你可以这样做:

 tf.contrib.layers.dense_to_sparse(sparse_matmul_result)

以上是关于如何在TensorFlow中执行稀疏矩阵*稀疏矩阵乘法?的主要内容,如果未能解决你的问题,请参考以下文章

稀疏矩阵与密集矩阵乘法 C++ Tensorflow

来自密集张量 Tensorflow 的稀疏张量(矩阵)

将密集向量与 Tensorflow 中稀疏矩阵的每一行相乘

稀疏矩阵及其压缩格式

在 Python 中对稀疏矩阵执行分解

使用 sklearn 对大型稀疏矩阵执行 PCA