[Pytorch] Boolean value of Tensor with more than one value is ambiguous 에러를 해결하자.

pytorch 코드를 사용하다 위와같은 에러를 만난 경우가 있다. 

 

가끔 만나는 에러라 뭐였지 하고 검색했다가 아 이거였지 하는 내모습에.. 

이번에는 기록을 해 둔다..

---

보통 내 경우는 loss 함수를 선언하고 해당 객체에 logit 과 label을 넣어줘야 하는데 

loss 를 선언하며 파라미터로 logit과 label을 넣어버려서 이런 경우가 발생했다.

 

즉,

# error 상황
loss = loss_func(logit, label)

# 올바른 코드
loss = loss_func()(logit, label)

 

객체 생성 후 값을 넣어주면 해당 에러는 해결 된다.