(pytorch) torch.permute

torch.permute torch.permute — PyTorch 1.12 documentation torch.permute torch. permute ( input , dims ) → Tensor Returns a view of the original tensor input with its dimensions permuted. Parameters input ( Tensor ) – the input tensor. dims ( tuple of python:ints ) – The desired ordering of dimensions Example >>> x = torch . randn ( 2 , 3 , 5 ) >>> x . size... pytorch.org 기능적으로 numpy의 transpose 함수와 동일하다. torch.transpose 함수의 경우에는 입력으로 2개의 dimension index만 받는 데 반해 torch.permute 함수의 경우에는 입력으로 여러 개의 d...