code: def set_batch_shape(tensor, batch_shape):
      return tf.broadcast_to(tensor, batch_shape + list(tensor.shape))
  
file_path: graphics/tensorflow_graphics/rendering/tests/rasterization_test_utils.py
imports: import os
  import random
  import numpy
  import tensorflow
  from tensorflow_graphics.geometry.transformation import look_at
  from tensorflow_graphics.rendering.camera import perspective
  from tensorflow_graphics.util import shape
label_desc: Use image node_id 6 for calling a global function with example usage: set_batch_shape(tensor, batch_shape) and returns: tf