def ch_aranged_batch(time_aranged_batch):
    """Generates a batch of size 1, where the feature matrix has 50 repeated
    columns of integers arranged between 0 and 63.
    """
    X, y = time_aranged_batch
    return X.transpose(1, 2), y
Generates a batch of size 1, where the feature matrix has 50 repeated
    columns of integers arranged between 0 and 63.
    
test/unit_tests/augmentation/test_transforms.py