matrix_transpose

paddle. matrix_transpose ( x, name=None ) [源代码]

对输入张量 x 的最后两个维度进行转置。

注解

如果 n 是 x 的维数，则 paddle.matrix_transpose(x) 等同于 x.transpose([0, 1, ..., n-2, n-1]) 。

参数

  • x (Tensor) - 输入 Tensor，数据类型为 float32，float64，int32 或者 int64。

  • name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回

Tensor，数据类型与 x 相同。

代码示例

>>> import paddle
>>> x = paddle.ones(shape=[2, 3, 5])
>>> x_transposed = paddle.matrix_transpose(x)
>>> print(x_transposed.shape)
[2, 5, 3]