ValueError: only one element tensors can be converted to Python scalars

ValueError: only one element tensors can be converted to Python scalars

 

위와 같은 에러가 발생할 경우

 

torch.tensor 를 사용하지 말고 torch.stack 을 사용하면 해결된다.

 

# As is
torch.tensor(y_pred) 

# To be
torch.stack(y_pred, dim=0)