Method to

python/dgl/frame.py:944–965

Return a new frame with columns copy to the targeted device (cpu/gpu). Parameters ---------- device : Framework-specific device context object The context to move data to. kwargs : Key-word arguments. Key-word arguments fed to the framework co

(self, device, **kwargs)

        return subf

    def to(self, device, **kwargs):  # pylint: disable=invalid-name
        """Return a new frame with columns copy to the targeted device (cpu/gpu).

        Parameters
        ----------
        device : Framework-specific device context object
            The context to move data to.
        kwargs : Key-word arguments.
            Key-word arguments fed to the framework copy function.

        Returns
        -------
        Frame
            A new frame
        """
        newframe = self.clone()
        new_columns = {
            key: col.to(device, **kwargs)
            for key, col in newframe._columns.items()
        }
        newframe._columns = new_columns
        return newframe

    def __repr__(self):
        return repr(dict(self))