본문 바로가기

AI/파이토치

torch.Tensor와 torch.tensor 차이

- "torch.Tensor"
    - 클래스 (Class)
    - int 입력 시 float로 변환
    - torch 데이터 입력 시 입력받은 데이터의 메모리 공간을 사용
    - list, numpy 데이터 입력 시 입력받은 데이터를 복사하여
      새롭게 torch.Tensor를 만든 후 사용
- "torch.tensor"
    - 함수 (Function)
    - int 입력시 int 그대로 입력
    - 입력 받은 데이터를 새로운 메모리 공간으로 복사 후 사용

 

 

우선 torch.tensor함수이다.  또한 입력 받는 데이터를 복사한 후 새로운 메모리 공간에 할당하게 된다. 

반면 torch.Tensor클래스이고 입력 받은 데이터의 메모리 공간을 사용한다.(주소값을 참조) 

 

** B = torch.Tensor(A) 의 경우 텐서 객체 B를 수정하게 되면 A의 값도 바뀐다.

 

'AI > 파이토치' 카테고리의 다른 글

nn.Module  (0) 2022.01.31
파이토치 함수들  (0) 2022.01.31