【发布时间】:2018-08-14 11:41:05
【问题描述】:
目前我正在尝试使用 TensorFlow 计算矩阵乘法,其中两个矩阵都是稀疏形式。我尝试使用tf.sparse_matmul(),但在这种情况下,我不得不转换回密集形式,这破坏了我的项目目标。请建议我在这种情况下我能做什么。
【问题讨论】:
标签: python tensorflow matrix
目前我正在尝试使用 TensorFlow 计算矩阵乘法,其中两个矩阵都是稀疏形式。我尝试使用tf.sparse_matmul(),但在这种情况下,我不得不转换回密集形式,这破坏了我的项目目标。请建议我在这种情况下我能做什么。
【问题讨论】:
标签: python tensorflow matrix
你试过tf.sparse_matmul吗?
sparse_matumul_result = tf.sparse_matmul(a, b)
如果你也需要sparse_matmul_result 作为稀疏,你可以这样做:
tf.contrib.layers.dense_to_sparse(sparse_matmul_result)
【讨论】: