import torch


def any(input_, axis=None, keepdims=False):
    """Wrapper of `torch.any`.

    Parameters
    ----------
    input_ : DTensor
        Input tensor.
    axis : None or int or tuple of ints, optional
        Axis or axes to operate on, by default None
    keepdims : bool, optional
        If true, the axes along which the operation performed are left to size
        one, by default False

    Returns
    -------
    DTensor or bool:
    """
    if axis is None:
        return torch.any(input_._data)
    return torch.any(input_._data, dim=axis, keepdim=keepdims)