Tensorflow中的批量稀疏矩阵乘法

Question

有没有办法在Tensorflow中执行批量稀疏矩阵乘法？这些是我试图成倍增加的形状：

[n，m，i，j] x [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
)