Function tensor

python/tvm/runtime/_tensor.py:330–353

Create an tensor from source arr. Parameters ---------- arr : numpy.ndarray The array to be copied from device : Device, optional The device to create the array mem_scope : Optional[str] The memory scope of the array Returns ------- ret : T

(arr, device=None, mem_scope=None)

def tensor(arr, device=None, mem_scope=None):
    """Create an tensor from source arr.

    Parameters
    ----------
    arr : numpy.ndarray
        The array to be copied from

    device : Device, optional
        The device to create the array

    mem_scope : Optional[str]
        The memory scope of the array

    Returns
    -------
    ret : Tensor
        The created array
    """
    device = device or cpu()

    if not isinstance(arr, np.ndarray | Tensor):
        arr = np.asarray(arr)
    return empty(arr.shape, arr.dtype, device, mem_scope).copyfrom(arr)


def cpu(dev_id=0):