TensorFlow函数:tf.matrix_transpose

2018-11-01 18:35 更新

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.
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号