o yi@s*ddlmZdedededdfddZdS) )Tensorpredstarget num_outputsreturnNcCsz|jdks |jdkrtd|jd|jd|dkr |jdks+|dkr9||jdkr;td|d|jdddSdS)zKCheck that predictions and target have the correct shape, else raise error.zWExpected both predictions and target to be either 1- or 2-dimensional tensors, but got z and .zPExpected argument `num_outputs` to match the second dimension of input, but got N)ndim ValueErrorshape)rrrr \/home/ubuntu/.local/lib/python3.10/site-packages/torchmetrics/functional/regression/utils.py _check_data_shape_to_num_outputss(r)torchrintrr r r rs