您可以通过简单的重塑来完成此操作.对于两个矩阵乘法中的第一个,你有k*d,长度为d的矢量与dot product.
这应该是接近的:
temp = tf.matmul(E1,tf.reshape(Wddk,[d,d*k])) result = tf.matmul(E2,tf.reshape(temp,[d,k]))