텐서(Tensor)의 자료형/타입을 어떻게 변경하나요?

[9bow (박정환)]

공식 홈페이지와 StackOverflow 등에서 자주 보이는 질문과 답변을 번역하고 있습니다.

다음 링크에서 원문을 함께 찾아보실 수 있습니다.

• 원문: How to cast a tensor to another type? - PyTorch Forums

---

질문

• Float로 선언한 Tensor의 자료형을 Double로 변경하고 싶습니다. 어떻게 해야하나요?
  (= FloatTensor를 어떻게 DoubleTensor로 변경할 수 있나요?)

[9bow (박정환)]

답변

• tensor.type(torch.DoubleTensor) 또는 tensor.type('torch.DoubleTensor')로 변경할 수 있습니다.
• 예시 코드는 아래와 같습니다.

  double_tensor = float_tensor.type(torch.DoubleTensor)
  

  또는

  double_tensor = float_tensor.type('torch.DoubleTensor')
  

더 알아보기

• torch.Tensor의 여러가지 자료형들(dtype): Tensor Attributes — PyTorch 2.6 documentation
• 각 dtype들과 이에 해당하는 CPU / GPU 텐서들: torch.Tensor — PyTorch 2.6 documentation
• torch.type() 문서: torch.Tensor — PyTorch 2.6 documentation