当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch transpose用法及代码示例


本文简要介绍python语言中 torch.transpose 的用法。

用法:

torch.transpose(input, dim0, dim1) → Tensor

参数

  • input(Tensor) -输入张量。

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

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

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

生成的 out 张量与 input 张量共享其底层存储,因此更改其中一个的内容会更改另一个的内容。

例子:

>>> 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.org大神的英文原创作品 torch.transpose。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。