code: def generate(self, z):
      return nn.sigmoid(self.decoder(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 3 for calling the VAE obj's underlying member method code with example usage: obj.generate(z) and returns: nn