파이토치 텐서 모양 바꾸기 tensor reshape

파이토치 텐서 모양 바꾸기 tensor reshape¶

• 텐서 예시 생성

In [2]:

import torch
tensor = torch.arange(9)
tensor

Out[2]:

tensor([0, 1, 2, 3, 4, 5, 6, 7, 8])

• 텐서 모양 변경

In [3]:

tensor.reshape(3, 3)

Out[3]:

tensor([[0, 1, 2],
        [3, 4, 5],
        [6, 7, 8]])