import torch


def rot90(input_, k=1, axes=(0, 1)):
    """Wrapper of `torch.rot90`

    Parameters
    ----------
    input_ : DTensor
        Input dense tensor.
    k : int, optional
        Number of rotation times, by default 1
    axes : tuple, optional
        The axes in which the input is rotated, by default (0, 1)
    """
    return torch.rot90(input_._data, k, dims=axes)