class_docstr: Move tensor to device.
  
  Parameters
  ----------
  device: torch.device
      Device to move to.
  non_blocking: bool
      Asynchronous if copying from CPU to GPU.
code: def __init__(self, device, non_blocking=True):
      super().__init__()
      self.device = device
      # assert isinstance(device, ch.device), \
      #    f'Make sure device is a ch.device (not a {type(device)})'
      self.non_blocking = non_blocking
  
imports: import torch
  import numpy
  from typing import Callable, Optional, Tuple
  from ..pipeline.allocation_query import AllocationQuery
  from ..pipeline.operation import Operation
  from ..pipeline.state import State
  from dataclasses import replace
label_desc: Use image node_id 1 to create a new ToDevice object from inherited base classes: Operation with example: obj = ToDevice(device, non_blocking)