파이토치 텐서 슬라이스와 인덱싱 pytorch tensor slice and indexing

파이토치 텐서 슬라이스와 인덱싱 pytorch tensor slice and index¶

In [1]:

import torch
import numpy as np

• 예시 tensor 생성

In [2]:

# 예시 tensor 생성
tensor = torch.arange(9)
tensor = tensor.reshape(3, 3)
tensor

Out[2]:

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

• 1번 위치 텐서 인덱싱

In [3]:

tensor[1]

Out[3]:

tensor([3, 4, 5])

• 2차원 인텍싱

In [4]:

tensor[1][1], tensor[1, 1]

Out[4]:

(tensor(4), tensor(4))

• 슬라이싱

In [6]:

tensor[1:]

Out[6]:

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

In [7]:

tensor[1:, 1:]

Out[7]:

tensor([[4, 5],
        [7, 8]])

In [8]:

tensor[1:, 1:2]

Out[8]:

tensor([[4],
        [7]])