code: def configure_task_integ_grads_embeddings(task):
      r"""
      Wraps Pytext's DocNN model embedding with `IntegratedGradientsEmbedding` for
      a given input task.
      IntegratedGradientsEmbedding allows to perform baseline related operations
  
      Args
  
          task: DocNN task reference
  
      Returns
  
          integrated_gradients_embedding_lst: The embedding layer which contains
                      IntegratedGradientsEmbedding as a wrapper over the original
                      embeddings of the model
  
      """
      integrated_gradients_embedding_lst = (
          configure_model_integ_grads_embeddings(task.model)
      )
      task.model.embedding = integrated_gradients_embedding_lst
      return integrated_gradients_embedding_lst[0]
  
imports: from collections import defaultdict
  import torch
  from pytext.models.embeddings.dict_embedding import DictEmbedding
  from pytext.models.embeddings.word_embedding import WordEmbedding
  from pytext.models.model import EmbeddingBase, EmbeddingList
label_desc: Use image node_id 1 for calling a global function with example usage: configure_task_integ_grads_embeddings(task) and returns: integrated_gradients_embedding_lst