Function cumsum

monai/transforms/utils_pytorch_numpy_unification.py:319–335

`np.cumsum` with equivalent implementation for torch. Args: a: input data to compute cumsum. axis: expected axis to compute cumsum. kwargs: if `a` is PyTorch Tensor, additional args for `torch.cumsum`, more details: https://pytorch.org/docs/stable/genera

(a: NdarrayOrTensor, axis=None, **kwargs)

def cumsum(a: NdarrayOrTensor, axis=None, **kwargs) -> NdarrayOrTensor:
    """
    `np.cumsum` with equivalent implementation for torch.

    Args:
        a: input data to compute cumsum.
        axis: expected axis to compute cumsum.
        kwargs: if `a` is PyTorch Tensor, additional args for `torch.cumsum`, more details:
            https://pytorch.org/docs/stable/generated/torch.cumsum.html.

    """

    if isinstance(a, np.ndarray):
        return np.cumsum(a, axis)  # type: ignore
    if axis is None:
        return torch.cumsum(a[:], 0, **kwargs)
    return torch.cumsum(a, dim=axis, **kwargs)


def isfinite(x: NdarrayOrTensor) -> NdarrayOrTensor: