Method copyto

python/tvm/runtime/_tensor.py:229–245

Copy array to target Parameters ---------- target : Tensor The target array to be copied, must have same shape as this array. mem_scope : Optional[str] The memory scope of the array.

(self, target, mem_scope=None)

        return np_arr

    def copyto(self, target, mem_scope=None):
        """Copy array to target

        Parameters
        ----------
        target : Tensor
            The target array to be copied, must have same shape as this array.

        mem_scope : Optional[str]
            The memory scope of the array.
        """
        if isinstance(target, Tensor):
            return self._copyto(target)
        if isinstance(target, tvm_ffi.core.Device):
            res = empty(self.shape, self.dtype, target, mem_scope)
            return self._copyto(res)
        raise ValueError(f"Unsupported target type {type(target)}")

    def _copyto(self, target_nd):
        """Internal function that implements copy to target ndarray."""