code: def Exp():
      """Returns a layer that computes the element-wise exponential of a tensor."""
      return Fn("Exp", lambda x: jnp.exp(x))
  
imports: from trax import fastmath
  from trax.fastmath import numpy
  from trax.layers import base
  from trax.layers.assert_shape import assert_shape
  from trax.layers.base import Fn
label_desc: Use image node_id 13 for calling a global function with example usage: Exp() and returns: Fn