如何在張量流中的稀疏和張量之間進行元素相乘？

Question

Tensorflow具有稀疏到密集矩陣乘法的實現tf.sparse_tensor_dense_matmul，但它對稀疏元素乘法（兩個張量具有相同形狀）是否稀疏？我想避免將我的稀疏張量轉換爲密集的張量，因爲它不適合記憶。

Answer 1

我不相信有一個內置的功能，但你可以用手比較容易地做到這一點，至少如果你不打算支持廣播。如果分別爲x和y。你的疏與密張量，

res = tf.SparseTensor(x.indices, tf.gather_nd(y, x.indices) * x.values, x.dense_shape) 

您可能還需要檢查的y和x的形狀相同乘以他們。