Method to_device

examples/model_selection/Trails/singa_pkg_code/tensor.py:283–290

Move the tensor data onto a given device. Args: device: a swig Device converted from CudaGPU or CppCPU or OpenclGPU

(self, device)

        return t

    def to_device(self, device):
        '''Move the tensor data onto a given device.

        Args:
            device: a swig Device converted from CudaGPU or CppCPU or OpenclGPU
        '''
        self.data.ToDevice(device)
        self.device = device

    def to_host(self):
        '''Move the tensor data onto the default host CppCPU device.