torchmetrics.utilities.data select_topk torchmetrics.utilities.data.select_topk to_categorical torchmetrics.utilities.data.to_categorical to_onehot torchmetrics.utilities.data.to_onehot