code: def __call__(self, z):
      z = nn.Dense(500, name="fc1")(z)
      z = nn.relu(z)
      z = nn.Dense(784, name="fc2")(z)
      return z
  
file_path: jax/jax/experimental/jax2tf/tests/flax_models/vae.py
imports: from flax import linen
  import jax.numpy
  from jax import random
label_desc: Use image node_id 1 for calling the Decoder obj's underlying member method code with example usage: obj.__call__(z) and returns: z