code: def map_vqa_split(batch):
      """Split answer into decoder_input and decoder_output."""
  
      full_tokens = batch.pop("answer")
      batch["decoder_input_tokens"] = full_tokens[..., :-1]
      batch["decoder_target_tokens"] = full_tokens[..., 1:]
      return batch
  
file_path: scenic/scenic/projects/knowledge_visual_language/data/vqa_dataset.py
imports: import functools
  from typing import Optional
  from absl import logging
  from flax import jax_utils
  import jax.numpy
  import ml_collections
  from scenic.dataset_lib import dataset_utils
  from scenic.dataset_lib import datasets
  from scenic.dataset_lib.big_transfer import bit
  from scenic.dataset_lib.big_transfer import builder
  from scenic.dataset_lib.big_transfer import registry
  from scenic.dataset_lib import web_image_text_dataset
  from scenic.projects.knowledge_visual_language.data import data_utils
  import tensorflow
label_desc: Use image node_id 2 for calling a global function with example usage: map_vqa_split(batch) and returns: batch