class_docstr: A device which executes operations in parallel.
class_docstr_tok: ["A","device","which","executes","operations","in","parallel","."]
docstr: Helper to unpack a single tensor.
docstr_tok: ["Helper","to","unpack","a","single","tensor","."]
code: def _unpack_tensor(self, parallel_tensor):
      """Helper to unpack a single tensor."""
      if not isinstance(
          parallel_tensor,
          (
              tensor_lib.Tensor,
              composite_tensor.CompositeTensor,
              variables.Variable,
          ),
      ):
          raise ValueError(
              "Expected a tensor, got {}.".format(parallel_tensor)
          )
      with ops.device(self._name):
          return tpu_ops.tpu_replicated_output(
              parallel_tensor, num_replicas=len(self.components)
          )
  
imports: import threading
  import weakref
  from tensorflow.python import _pywrap_parallel_device
  from tensorflow.python.distribute import device_util
  from tensorflow.python.eager import context
  from tensorflow.python.framework import composite_tensor
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.tpu.ops import tpu_ops
  from tensorflow.python.util import nest
  from tensorflow.python.util import variable_utils
label_desc: Use image node_id 4 for calling the ParallelDevice obj's underlying member method code with example usage: obj._unpack_tensor(parallel_tensor) and returns: tpu_ops