code: def random_indices(
      batch_size: Union[Tensor, int],
      n: Union[Tensor, int],
      n_max: Union[Tensor, int],
  ) -> Tensor:
      """Creates `batch_size * n` random indices that run from `0` to `n_max`.
  
      Args:
          batch_size: Number of items in each batch
          n: Number of random indices in each example
          n_max: Maximum index (excluded)
  
      Returns:
          A uniformly distributed integer tensor of indices
      """
      return tf.random.uniform(
          shape=(batch_size, n), maxval=n_max, dtype=tf.int32
      )
  
imports: import tensorflow
  from tensorflow import Tensor
  from typing import Union, Optional
label_desc: Use image node_id 1 for calling a global function with example usage: random_indices(batch_size, n, n_max) and returns: tf