def fold_split(self, random_seed=None):
    '\n        Splitting the folds.\n\n        Args:\n            random_seed: Random seed for reproducibility\n\n        Returns:\n            tensor containing indices for folds, where dim=0 is the fold number\n\n        '
    if (random_seed is not None):
        torch.manual_seed(random_seed)
    fold_idx = torch.randperm(self.dataset.__len__())
    fold_idx = fold_idx[:self.folded_size].view((- 1), self.fold_size)
    return fold_idx
Splitting the folds.

Args:
    random_seed: Random seed for reproducibility

Returns:
    tensor containing indices for folds, where dim=0 is the fold number
def fold_split(self, random_seed=None):
    '\n        Splitting the folds.\n\n        Args:\n            random_seed: Random seed for reproducibility\n\n        Returns:\n            tensor containing indices for folds, where dim=0 is the fold number\n\n        '
    if (random_seed is not None):
        torch.manual_seed(random_seed)
    fold_idx = torch.randperm(self.dataset.__len__())
    fold_idx = fold_idx[:self.folded_size].view((- 1), self.fold_size)
    return fold_idx