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

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

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


질문

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

답변

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

더 알아보기

1개의 좋아요