Tensorflow中批量稀疏矩陣乘法

Question

在Tensorflow中執行批量稀疏矩陣乘法嗎？這些是我試圖乘以形狀：

[N，M，I，J]×[N，M，J，K] = [N，M，I，K]

因此，有是雙方的批次組件，並且每個2D內部矩陣對應相應地相乘。目前實現的功能有沒有辦法？

謝謝。

Answer 1

你沒有提到你是否嘗試過任何東西，所以我現在就發佈這個。

從tensorflow文檔中，tf.matmul支持批處理 乘法。

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 
)