인프런 커뮤니티 질문&답변

wlghks94498님의 프로필 이미지
wlghks94498

작성한 질문수

처음하는 딥러닝과 파이토치(Pytorch) 부트캠프 (쉽게! 기본부터 챗GPT 핵심 트랜스포머까지) [데이터분석/과학 Part4]

파이토치(PyTorch) 기본 문법 이해: 텐서 조건 연산과 저장

섹션2 코드질문

작성

·

293

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() 으로 사용하시면 됩니다. 관련 자료도 업데이트하여 업로드하였습니다.

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

wlghks94498님의 프로필 이미지
wlghks94498

작성한 질문수

질문하기