class_docstr: VAE Encoder.
code: def __call__(self, x):
      x = nn.Dense(500, name="fc1")(x)
      x = nn.relu(x)
      mean_x = nn.Dense(self.latents, name="fc2_mean")(x)
      logvar_x = nn.Dense(self.latents, name="fc2_logvar")(x)
      return mean_x, logvar_x
  
file_path: flax/examples/vae/models.py
imports: from flax import linen
  from jax import random
  import jax.numpy
label_desc: Use image node_id 1 for calling the Encoder obj's underlying member method code with example usage: obj.__call__(x) and returns: mean_x,  logvar_x