o .wi  @s2ddlmZ d dededededdf d d ZdS) )TensorFpredstarget num_outputsallow_1d_reshapereturnNcCs|jdks |jdkrtd|jd|jdd}|s,|dko+|jdkp*|jddk }|dko;|jdko;||jdk}|s@|rNtd|d|jdddS) aCheck that predictions and target have the correct shape, else raise error. Args: preds: Predicted tensor target: Ground truth tensor num_outputs: Number of outputs in multioutput setting allow_1d_reshape: Allow that for num_outputs=1 that preds and target does not need to be 1d tensors. Instead code that follows are expected to reshape the tensors to 1d. zWExpected both predictions and target to be either 1- or 2-dimensional tensors, but got z and .FzPExpected argument `num_outputs` to match the second dimension of input, but got N)ndim ValueErrorshape)rrrrcond1cond2re/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/torchmetrics/functional/regression/utils.py _check_data_shape_to_num_outputss& " r)F)torchrintboolrrrrrs