Tensorflow中批量稀疏矩阵乘法
问题描述:
在Tensorflow中执行批量稀疏矩阵乘法吗?这些是我试图乘以形状:Tensorflow中批量稀疏矩阵乘法
[N,M,I,J]×[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
)