작성
·
272
0
Binary Classfication 딥러닝 적용해보기 강의 19:13초에서,
y_pred_list.extend(y_test_pred.squeeze().detach().tolist())
를 실행한뒤, y_pred_list = torch.tensor(y_pred_list).unsqueeze(1)을 실행하는데,
y_pred_list의 size() 를 y_test_tensor의 size()와 맞추는거면
predlist.extend(y_test_pred.detach().tolist())
를 실행한뒤, y_pred_list = torch.tensor(y_pred_list)) 을 실행해도 같은 건가요?
답변 1
0
안녕하세요. 답변도우미입니다.
아핫 좀더 공식적으로 shape 을 만드는 부분을 보여드리려 하고, 관련 함수도 좀더 익숙해질겸 위 코드를 작성하였고요. 말씀하신대로 squeeze(), unsqeeze() 안해도 됩니다.
감사합니다.