import torch


def triu_indices(n, m=0, k=0):
    """Wrapper of `torch.triu_indices`.

    Parameters
    ----------
    n : int
        Number of row of output tensor.
    m : int, optional
        Number of column of output tensor, by default 0
    offset : int, optional
        Offset to main diagonal, by default 0
    """
    ret = torch.triu_indices(row=m, col=m, offset=k)
    return tuple([index for index in ret])