code: def _reshape_tensors(tensors, shape):
      """Reshape tensors flattened by _flatten_tensors.
  
      Args:
        tensors: list of `tf.Tensor` of identical length 1D tensors.
        shape: list of integers describing the desired shape.  Product of
          the elements must equal the length of each tensor.
  
      Returns:
        list of `tf.Tensor` which are the reshaped inputs.
      """
      reshaped = []
      for t in tensors:
          with ops.colocate_with(t):
              reshaped.append(array_ops.reshape(t, shape))
      return reshaped
  
imports: import collections
  import math
  from tensorflow.python.framework import device
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nccl_ops
label_desc: Use image node_id 2 for calling a global function with example usage: _reshape_tensors(tensors, shape) and returns: reshaped