Python PyTorch transpose用法及代碼示例

本文簡要介紹python語言中 torch.transpose 的用法。

用法:

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

參數：

• input(Tensor) -輸入張量。

• dim0(int) -要轉置的第一個維度

• dim1(int) -要轉置的第二個維度

返回一個張量，它是 input 的轉置版本。給定的維度dim0 和dim1 被交換。

生成的 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() 。