テンソルフローの疎なテンソルと密なテンソルの間の要素的な乗算を行う方法は？

Question

Tensorflowは、疎と高密度のマトリックス乗算の実装tf.sparse_tensor_dense_matmulを持っていますが、それは密度の高い要素ごとの乗算（2つのテンソルは同じ形をしています）には疎ですか？私はメモリに収まらないので、疎なテンソルを密なテンソルに変換しないでください。

Answer 1

私は組み込み関数があるとは思っていませんが、少なくともあなたが放送をサポートするつもりがない場合は、比較的簡単に手で行うことができます。 xおよびyがresp。あなたの粗密テンソル、

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

はまた、それらを掛ける前にyとxの形状が同じであることを確認する必要があります。