code: def encode(self, scope, x):
      return nn.dense(scope, x, self.latents, bias=False)
  
file_path: flax/tests/core/design/core_tied_autoencoder_test.py
imports: from dataclasses import dataclass
  import jax
  from absl.testing import absltest
  from jax import numpy
  from jax import random
  from flax.core import init, lift, nn, unfreeze
label_desc: Use image node_id 2 for calling the TiedAutoEncoder obj's underlying member method code with example usage: obj.encode(scope, x) and returns: nn