快捷方式

torch.transpose

torch.transpose(input, dim0, dim1) Tensor

返回一个张量,它是 input 的转置版本。给定的维度 dim0dim1 被交换。

如果 input 是一个带步长的张量,则生成的 out 张量与其底层存储共享 input 张量,因此更改一个的内容将更改另一个的内容。

如果 input 是一个 稀疏张量,则生成的 out 张量input 张量共享底层存储。

如果 input 是一个具有压缩布局的 稀疏张量 (SparseCSR、SparseBSR、SparseCSC 或 SparseBSC),则参数 dim0dim1 必须都是批处理维度,或者必须都是稀疏维度。稀疏张量的批处理维度是出现在稀疏维度之前的维度。

注意

交换 SparseCSRSparseCSC 布局张量的稀疏维度的转置将导致布局在两种选项之间变化。` SparseBSR` 或 SparseBSC 布局张量的稀疏维度的转置同样会生成具有相反布局的结果。

参数
  • input (Tensor) – 输入张量。

  • dim0 (int) – 要转置的第一个维度

  • dim1 (int) – 要转置的第二个维度

示例

>>> x = torch.randn(2, 3)
>>> x
tensor([[ 1.0028, -0.9893,  0.5809],
        [-0.1669,  0.7299,  0.4942]])
>>> torch.transpose(x, 0, 1)
tensor([[ 1.0028, -0.1669],
        [-0.9893,  0.7299],
        [ 0.5809,  0.4942]])

另请参见 torch.t().

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源