Rate this Page
★★★★★

torch.permute#

torch.permute(input,dims)→Tensor#

Returns a view of the original tensorinput with its dimensions permuted.

Parameters

  • input (Tensor) – the input tensor.

  • dims (tuple ofint) – The desired ordering of dimensions

Example

>>>x=torch.randn(2,3,5)>>>x.size()torch.Size([2, 3, 5])>>>torch.permute(x,(2,0,1)).size()torch.Size([5, 2, 3])