Function ensure_tuple_size

monai/utils/misc.py:181–191

Returns a copy of `tup` with `dim` values by either shortened or padded with `pad_val` as necessary.

(vals: Any, dim: int, pad_val: Any = 0, pad_from_start: bool = False)

def ensure_tuple_size(vals: Any, dim: int, pad_val: Any = 0, pad_from_start: bool = False) -> tuple:
    """
    Returns a copy of `tup` with `dim` values by either shortened or padded with `pad_val` as necessary.
    """
    tup = ensure_tuple(vals)
    pad_dim = dim - len(tup)
    if pad_dim <= 0:
        return tup[:dim]
    if pad_from_start:
        return (pad_val,) * pad_dim + tup
    return tup + (pad_val,) * pad_dim


def ensure_tuple_rep(tup: Any, dim: int) -> tuple[Any, ...]: