[PyTorch] Tensor와 tensor

PyTorch를 사용하다보면 Tensor가 있고, tensor가 있다.

그 둘의 차이는 무엇일까?

 

1. Tensor

torch.Tensor란 PyTorch의 자료구조 클래스이다.

tensor형 데이터를 담고 있는 클래스를 의미한다.

 

2. tensor

torch.tensor는 어떤 데이터를 tensor 자료형으로 만들어주는 함수이다.

(내부적으로는 데이터를 copy하는 구조)

즉, tensor를 통해서 어떠한 데이터를 Tensor 데이터 타입으로 만들어준다고 보면 된다.

 

 

다음 코드를 보면 의미를 이해하기 더 쉬울 것이다.

 

>>> a = torch.tensor([1])
>>> a
tensor([1])

>>> type(a)
torch.Tensor

 

 

아래 document를 살펴보면 Tensor 구조체에서 사용할 수 있는 다양한 함수들을 확인해 볼 수 있다!

 

torch.Tensor — PyTorch 1.10.1 documentation