Function concatenate

monai/transforms/utils_pytorch_numpy_unification.py:312–316

`np.concatenate` with equivalent implementation for torch (`torch.cat`).

(to_cat: Sequence[NdarrayOrTensor], axis: int = 0, out=None)

def concatenate(to_cat: Sequence[NdarrayOrTensor], axis: int = 0, out=None) -> NdarrayOrTensor:
    """`np.concatenate` with equivalent implementation for torch (`torch.cat`)."""
    if isinstance(to_cat[0], np.ndarray):
        return np.concatenate(to_cat, axis, out)  # type: ignore
    return torch.cat(to_cat, dim=axis, out=out)  # type: ignore


def cumsum(a: NdarrayOrTensor, axis=None, **kwargs) -> NdarrayOrTensor: