강의

멘토링

커뮤니티

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

김성웅님의 프로필 이미지
김성웅

작성한 질문수

[파이토치] 실전 인공지능으로 이어지는 딥러닝 - 기초부터 논문 구현까지

집 값 예측하기 - 다층 신경망(MLP)

코드 질문입니다.

작성

·

157

0

안녕하세요 강사님.

Test 모드에서 테스트하고 나온 예측값과 실제값을 numpy로 변경하는 이유가 궁금합니다.

답변 1

1

딥러닝호형님의 프로필 이미지
딥러닝호형
지식공유자

안녕하세요!

 

좋은 지적 감사합니다.

코드 작성할 때 mean_squared_error가 텐서를 못 받는다고 생각했던 것 같습니다 ^^;;;

확인해보니 해당 함수가 array 체크를 자체적으로 해주어 텐서 계산도 가능하네요. 하지만 gpu용 텐서는 계산이 불가능하니 cpu용 텐서나 넘파이 배열로 바꿔주셔야 합니다!

 

감사합니다.

 

 

 
김성웅님의 프로필 이미지
김성웅
질문자

친절한 설명 감사드립니다!!

김성웅님의 프로필 이미지
김성웅

작성한 질문수

질문하기