파이토치 텐서의 기본 구성요소

파이토치(Pytorch)의 기본 데이터 구조인 텐서(Tensor)는 데이터와 gradient계산을 위한 메타데이터를 포함하고 있다.

기본적으로 텐서는 다음과 같은 주요 속성을 갖는다.

 

 

1. data

텐서의 주요 구성 요소로, 실제 수치 정보를 저장한다. 데이터는 다양한 차원을 가질 수 있으며, 신경망에서는 주로 벡터, 행렬 또는 더 높은 차원의 배열로 사용한다.

 

2. dtype

텐서에 저장된 데이터 타입을 정의한다. 예를 들어, 'torch.float32'; 'torch.int64'등이 있다.

 

3. device

텐서가 어떤 장치(CPU,GPU)에 할당되어 있는지를 나타낸다.

 

4. requires_grad

이 속성이 'True'로 설정되어 있으면, 텐서에 대한 모든 연산은 자동 미분 시스템에 의해서 추적되게 된다. 이를 통해 gradient가 자동으로 계산되며, 이는 학습과정에서 매우 중요한 요소이다.

 

5. grad

requires_grad가 'True'로 설정된 텐서에 대해 연산이 수행될 때, Pytorch는 자동으로 gradient를 계산하고 이를 grad속성에 저장한다. 이 gradient는 파라미터의 최적화에 사용된다.

 

6.shape

텐서의 차원을 나타내며, 각 차원의 크기(Size)를 포함한다.