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

问题描述 投票:0回答:1

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

python tensorflow matrix
1个回答
0
投票

你试过tf.sparse_matmul吗?

 sparse_matumul_result = tf.sparse_matmul(a, b)

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

 tf.contrib.layers.dense_to_sparse(sparse_matmul_result)
© www.soinside.com 2019 - 2024. All rights reserved.