peal.global_utils.move_to_device¶

peal.global_utils.move_to_device(X, device)[source]¶

Clone a tensor, or a list of tensors, onto device.

Parameters:

X (torch.Tensor or list of torch.Tensor) – The data to move.
device (torch.device or str) – Target device.

Returns:

Copies, so the originals keep their device and graph.

Return type:

torch.Tensor or list of torch.Tensor