Size

Size 란?

Size() 함수는 PyTorch 텐서의 각 차원 크기를 반환하는 메서드이다. 이를 통해서 텐서의 shape을 쉽게 확인할 수 있다.

기본 문법

텐서tensor.size(dim=None)

• 이때 특정 차원의 크기를 반환하려면 해당 차원의 인덱스를 전달하면 된다.

예시

import torch

# 예시 텐서 생성
tensor = torch.tensor([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

# 텐서의 전체 크기 확인
print(tensor.size())  # 출력: torch.Size([3, 3])

# 특정 차원의 크기 확인
print(tensor.size(0))  # 출력: 3 (첫 번째 차원의 크기)
print(tensor.size(1))  # 출력: 3 (두 번째 차원의 크기)