def _batch_graphs(graphs):
  """Concatenates a list of graphs along the batch dimension.

  Args:
    graphs (iterable): graphs to be concatenated.
  """
  result = graphs[0].clone()
  for idx in range(1, len(graphs)):
    result.append(graphs[idx])
  return result