dice_similarity_coefficient_with_logits(): RuntimeError: Boolean value of Tensor with more than one value is ambiguous#210