code: def _exponentially_smoothed(momentum, old, new):
      smoothed_value = momentum * old + (1 - momentum) * new
      return smoothed_value.astype(old.dtype)
  
imports: from trax.fastmath import numpy
  from trax.layers import base
label_desc: Use image node_id 1 for calling a global function with example usage: _exponentially_smoothed(momentum, old, new) and returns: smoothed_value