小编Abh*_*oke的帖子

在tensorflow tf.einsum()中可以用于矩阵乘法.例如,考虑我有两个张量A(dim=100,28,28,512)和B(dim=100,28,28,512).我tf.einsum('ijkm,ijkn>imn',A,B)用来沿第二和第三维乘以矩阵A和B. 此乘法的输出维数为(100,512,512).

在这种情况下,张量流如何处理梯度反向传播？