Function transpose

python/tvm/topi/transform.py:110–125

Permute the dimensions of an array. Parameters ---------- a : tvm.te.Tensor The tensor to be expanded. axes: tuple of ints, optional By default, reverse the dimensions. Returns ------- ret : tvm.te.Tensor

(a, axes=None)

def transpose(a, axes=None):
    """Permute the dimensions of an array.

    Parameters
    ----------
    a : tvm.te.Tensor
        The tensor to be expanded.

    axes: tuple of ints, optional
        By default, reverse the dimensions.

    Returns
    -------
    ret : tvm.te.Tensor
    """
    return cpp.transpose(a, axes)


def flip(a, axis=0):