Tensorflow具有稀疏到密集矩阵乘法的实现，但它是否具有稀疏到密集的元素乘法（两个具有相同形状的张量）？我想避免将我的稀疏张量转换为密集的张量，因为它不适合记忆 .
我不打算支持广播 . 如果 x 和 y 是分别的 . 你稀疏而密集的张量，
x
y
res = tf.SparseTensor(x.indices, tf.gather_nd(y, x.indices) * x.values, x.dense_shape)
您可能还需要在将它们相乘之前检查 y 和 x 的形状是否相同 .
1 回答
我不打算支持广播 . 如果
x和y是分别的 . 你稀疏而密集的张量，您可能还需要在将它们相乘之前检查
y和x的形状是否相同 .