Permutation Invariant Training (PIT) Module Interface torchmetrics.PermutationInvariantTraining Functional Interface torchmetrics.functional.permutation_invariant_training