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 
)