code: def _repeat_n(new_batch: int, data: jnp.ndarray) -> jnp.ndarray:
      """Create new batch dimension of size `new_batch` by repeating `data`."""
      return jnp.broadcast_to(data, (new_batch,) + data.shape)
  
file_path: acme/acme/agents/jax/mbop/ensemble.py
imports: import functools
  from typing import Any, Callable
  from acme.jax import networks
  import jax
  import jax.numpy
label_desc: Use image node_id 2 for calling a global function with example usage: _repeat_n(new_batch, data) and returns: jnp