def sample_block_indices_uniformly(n: int, total_num_blocks: int) -> list[int]:
    """
    Sample N block indices uniformly from the total number of blocks.
    """
    return [
        int(total_num_blocks / n * block_depth) - 1 for block_depth in range(1, n + 1)
    ]

    Sample N block indices uniformly from the total number of blocks.
    
segmentation_models_pytorch/encoders/timm_vit.py