Python Tensorflow中的批稀疏矩阵乘法

在Tensorflow中是否有执行批稀疏矩阵乘法的方法？这些是我试图复制的形状：

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

因此，两边都有一个批处理组件，每个2D内部矩阵对应该相应地相乘。对于当前实现的功能有什么方法吗

---

谢谢

你没有提到你是否尝试过任何东西，所以我现在就发布这个

从tensorflow文档中，支持批处理 乘法运算

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
)

---

你没有提到你是否尝试过任何东西，所以我现在就发布这个

从tensorflow文档中，支持批处理 乘法运算

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
)

---

这个答案可能不正确。文档中说稀疏矩阵的优化只适用于rand 2张量。这个答案可能不正确。文档中说稀疏矩阵的优化只适用于rand 2张量。