tf.matrix_transpose 函数

matrix_transpose(
    a,
    name='matrix_transpose'
)

定义在：tensorflow/python/ops/array_ops.py.

参考指南：数学函数>矩阵数学函数

转置张量 a 的最后两个维数.

例如：

# Matrix with no batch dimension.
# 'x' is [[1 2 3]
#         [4 5 6]]
tf.matrix_transpose(x) ==> [[1 4]
                                 [2 5]
                                 [3 6]]

# Matrix with two batch dimensions.
# x.shape is [1, 2, 3, 4]
# tf.matrix_transpose(x) is shape [1, 2, 4, 3]

注意,tf.matmul 提供 kwargs,它允许转置参数.这是以最小的成本完成的,并且比使用此函数更可取.例如：

# Good!  Transpose is taken at minimal additional cost.
tf.matmul(matrix, b, transpose_b=True)

# Inefficient!
tf.matmul(matrix, tf.matrix_transpose(b))

参数：

• a：张量,并且 rank >= 2.
• name：操作的名称(可选).

返回值：

该函数返回一个经过转置的批次矩阵张量.

可能引发的异常：

• ValueError：如果 a 确定是静态的,并且 rank < 2.