Skip to content

Commit

Permalink
Correction for docstring for multiclass stat scores tensor validation…
Browse files Browse the repository at this point in the history
… function (#1862)
  • Loading branch information
goldenfire6 authored Jun 28, 2023
1 parent cc427ba commit cb13126
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/torchmetrics/functional/classification/stat_scores.py
Original file line number Diff line number Diff line change
Expand Up @@ -254,7 +254,7 @@ def _multiclass_stat_scores_tensor_validation(
) -> None:
"""Validate tensor input.
- if target has one more dimension than preds, then all dimensions except for preds.shape[1] should match
- if preds has one more dimension than target, then all dimensions except for preds.shape[1] should match
exactly. preds.shape[1] should have size equal to number of classes
- if preds and target have same number of dims, then all dimensions should match
- if ``multidim_average`` is set to ``samplewise`` preds tensor needs to be atleast 2 dimensional in the
Expand Down

0 comments on commit cb13126

Please sign in to comment.