Function reshape

python/tvm/topi/transform.py:360–374

Reshape the array Parameters ---------- a : tvm.te.Tensor The tensor to be reshaped newshape : tuple of ints The new shape Returns ------- ret : tvm.te.Tensor

(a, newshape)

def reshape(a, newshape):
    """Reshape the array

    Parameters
    ----------
    a : tvm.te.Tensor
        The tensor to be reshaped
    newshape : tuple of ints
        The new shape

    Returns
    -------
    ret : tvm.te.Tensor
    """
    return cpp.reshape(a, newshape)


def squeeze(a, axis=None):