• 카테고리

    질문 & 답변
  • 세부 분야

    딥러닝 · 머신러닝

  • 해결 여부

    미해결

섹션2 코드질문

23.07.30 13:57 작성 조회수 246

0

# 동일 shape 은 당연히 Tensor 간 연산 가능

data1 = torch.torch.FloatTensor([ [1], [2], [3] ])

data2 = torch.FloatTensor([1, 1, 1])

data3 = data1 + data2

print (data1.shape, data2.shape, data3.shape)

print (data3)

 

이 코드에서 data1은

torch.FloatTensor이 아닌

torch.torch.FloatTensor를 사용하는 이유가 궁금합니다.

출력했을때는 같은결과가 나오는데 말이죠..

답변 1

답변을 작성해보세요.

0

안녕하세요. 답변 도우미입니다.

실수가 있었는데, 알려주셔서 감사합니다.

torch.FloatTensor() 로 사용하시면 됩니다. 유사 코드를 복사해서 작성하다보니, 실수가 있었던 것 같은데요. torch.torch.FloatTensor() 도 정상 동작하다보니, 실수를 발견하지 못했습니다. torch.FloatTensor() 가 일반적인 코드라서, torch.FloatTensor() 으로 사용하시면 됩니다. 관련 자료도 업데이트하여 업로드하였습니다.

불편을 드려 죄송합니다. 감사합니다.