# target.shape == output.shape criterion = torch objective function criterion(output[target == 원하는 값], target[target == 원하는 값])