import torch


def transpose(input_, axes=None):
    """Wrapper of `torch.transpose`.

    Parameters
    ----------
    input_ : DTensor
        Input tensor
    axes : list of ints, optional
       Axes along which the operation is performed, by default None
    """
    if axes is None:
        axes = (0, 1)
    return torch.transpose(input_, axes[0], axes[1])