파이토치(torch) 텐서 사이즈 보기

파이토치(torch) 텐서 사이즈 보기 (size(), shpae, dim())¶

In [1]:

import torch

In [2]:

# 예시 
x = torch.FloatTensor([
    [1, 2],
    [3, 4],
    [5, 6],
    [7, 8]
])

In [3]:

# size를 활용한 텐서 사이즈 보기
x.size()

Out[3]:

torch.Size([4, 2])

In [4]:

# shape 활용한 텐서 사이즈 보기
x.shape

Out[4]:

torch.Size([4, 2])

In [5]:

# dim을 활용한 텐서 차원 확인
x.dim()

Out[5]:

2

In [6]:

# 0 차원 사이즈 확인
x.size(0)

Out[6]:

4

In [7]:

# 1 차원 사이즈 확인
x.shape[0]

Out[7]:

4

• 딥러닝 모델에 데이터를 통과 시키기 위해서 텐서 사이즈 보기는 필수 인듯 ... 합니다...
• 네트워크 층을 통과 할때 마다 shape를 찍어보며 데이터 모양을 확인하고 왜 오류가 나는지 확인 합니다.
• 간단한 기능처럼 보이지만 ... 매우 중요하다고 생각하는.... 함수 인듯!!!