Edit
In Place Editing Advanced Editing

tf.matrix_transpose()

tf.matrix_transpose(a, name='matrix_transpose')

Transposes last two dimensions of tensor a.

For example:

# 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]
Args:
  • a: A Tensor with rank >= 2.
  • name: A name for the operation (optional).
Returns:

A transposed batch matrix Tensor.

Raises:
  • ValueError: If a is determined statically to have rank < 2.
doc_TensorFlow
doc_TensorFlow
2025-01-10 15:47:30
Comments
Leave a Comment

Please login to continue.