python - 为什么 tf.matmul 不能与转置张量一起使用？

Question

为什么不适tf.matmul用于转置张量？

transpose_b=True可以，但不行tf.transpose(inp)。

此屏幕截图是在 Colab 中制作的tensorflow-gpu==2.0.0-rc1：

标签： pythontensorflow

Answer 1

tf.transpose()perm如果您没有明确指定(permutation) 参数，则默认执行常规二维矩阵转置（它将 perm 参数设置为 input_tensor_rank-1） 。所以适当设置perm参数

inp_t = tf.transpose(inp, perm=[0,2,1])
y = tf.matmul(inp, x)
print(y)