[PyTorch] Boolean value of Tensor with more than one value is ambiguous

pred와 target값을 loss function에 전달하여 loss를 계산할때 아래와 같은 오류가 발생했다.

 

Boolean value of Tensor with more than one value is ambiguous

 

loss 계산  코드를 잘못 짠게 원인이었다.

 

def mse_loss(pred, target):
    loss = nn.MSELoss(pred, target)
    return loss

 

 위 코드를 아래와 같이 수정하니 오류 해결!

 

def mse_loss(pred, target):
    loss_func = nn.MSELoss()
    loss = loss_func(pred, target)
    return loss