有没有办法在Tensorflow中执行批量稀疏矩阵乘法？这些是我试图繁殖的形状：
[n，m，i，j] x [n，m，j，k] = [n，m，i，k]
因此，两侧都有一个批处理组件，每个2D内部矩阵对应相应地相乘 . 有没有办法使用当前实现的功能？
谢谢 .
你没有提到你是否尝试过任何东西，所以我现在就发布 .
从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 )
1 回答
你没有提到你是否尝试过任何东西，所以我现在就发布 .
从tensorflow文档中，tf.matmul支持批量乘法 .