code: def compute_full_importance(pi, mu):
      pimu = pi.all_prob.array / mu.all_prob.array
      # NaN occurs when inf/inf or 0/0
      pimu[np.isnan(pimu)] = 1
      pimu = np.nan_to_num(pimu)
      return pimu
  
imports: import copy
  from logging import getLogger
  import chainer
  from chainer import functions
  import numpy
  from chainerrl.action_value import SingleActionValue
  from chainerrl import agent
  from chainerrl import distribution
  from chainerrl import links
  from chainerrl.misc import async_
  from chainerrl.misc import copy_param
  from chainerrl.recurrent import Recurrent
  from chainerrl.recurrent import RecurrentChainMixin
  from chainerrl.recurrent import state_kept
  from chainerrl.recurrent import state_reset
label_desc: Use image node_id 2 for calling a global function with example usage: compute_full_importance(pi, mu) and returns: pimu