code: def reshape_from_matrix(output_tensor, orig_shape_list):
      """Reshapes a rank 2 tensor back to its original rank >= 2 tensor."""
      if len(orig_shape_list) == 2:
          return output_tensor
  
      output_shape = get_shape_list(output_tensor)
  
      orig_dims = orig_shape_list[0:-1]
      width = output_shape[-1]
  
      return tf.reshape(output_tensor, orig_dims + [width])
  
imports: import copy
  import json
  import math
  import numpy
  import tensorflow
  import functools
  from transformers.activations_tf import gelu
  from finetune.optimizers.recompute_grads import recompute_grad
  from finetune.nn.auxiliary import embed_position
label_desc: Use image node_id 18 for calling a global function with example usage: reshape_from_matrix(output_tensor, orig_shape_list) and returns: tf, output_tensor