Batch-Sparse-Matrix-Multiplikation in Tensorflow

Question

Gibt es eine Möglichkeit, Batch-Sparse-Matrix-Multiplikation in Tensorflow durchzuführen? Dies sind die Formen des ich versuche zu multiplizieren:

[n, m, i, j] x [n, m, j, k] = [n, m, i, k]

So gibt ist eine Stapelkomponente auf beiden Seiten, und jedes 2D innere Matrixpaar sollte entsprechend multipliziert werden. Gibt es einen Weg mit den aktuell implementierten Funktionen?

Danke.

Answer 1

Sie haben nicht erwähnt, wenn Sie etwas versucht haben, also werde ich das jetzt gerade posten.

Von der Tensorflow-Dokumentation unterstützt tf.matmul unterstützt Multiplikation.

matmul(
a, 
b, 
transpose_a=False, 
transpose_b=False, 
adjoint_a=False, 
adjoint_b=False, 
a_is_sparse=False, 
b_is_sparse=False, 
name=None 
)