code: def restore_batch_dims(
      tensor: type_alias.TensorLike, batch_shape: List[int]
  ) -> type_alias.TensorLike:
      """Unpack first dimension into batch_shape, preserving the rest of the dimensions."""
      return tf.reshape(
          tensor, batch_shape + tensor.shape.as_list()[1:]
      )
  
imports: from typing import List, Optional
  import tensorflow
  from tensorflow_graphics.util import type_alias
label_desc: Use image node_id 3 for calling a global function with example usage: restore_batch_dims(tensor, batch_shape) and returns: tf