在使用tf.matmul(x, y, transpose_b=False)做矩阵运算时,需要注意一下几点
1、如果transpose_b参数设置为False,那么x的最后一维要跟y的倒数第二维相等,即:
tf.matmul(x, y, transpose_b = False)
x.shape = [..., x1, x2]
y.shape = [..., y1, y2]
x2 = y1
2、如果transpose_b参数设置为True,那么x的最后一维要跟y的最后一维相等,transpose_b=True只是把第二个矩阵(这里就是y)的最后两维换了下,即:
tf.matmul(x, y, transpose_b = True)
x.shape = [..., x1, x2]
y.shape = [..., y1, y2]
x2 = y2