original_docstring: 
          Encoder that ouputs parameters for base distribution of z and context h for flows.
          
docstring: Encoder that ouputs parameters for base distribution of z and context h for flows.
docstring_tokens: Encoder that ouputs parameters for base distribution of z and context h for flows .
code: def encode(self, x):
          """
          Encoder that ouputs parameters for base distribution of z and context h for flows.
          """
  
          h = self.q_z_nn(x)
          h = h.view(-1, self.q_z_nn_output_dim)
          mean_z = self.q_z_mean(h)
          var_z = self.q_z_var(h)
  
          return mean_z, var_z, h
code_tokens: def encode ( self , x ) : h = self . q_z_nn ( x ) h = h . view ( - 1 , self . q_z_nn_output_dim ) mean_z = self . q_z_mean ( h ) var_z = self . q_z_var ( h ) return mean_z , var_z , h
short_docstring: Encoder that ouputs parameters for base distribution of z and context h for flows.
short_docstring_tokens: Encoder that ouputs parameters for base distribution of z and context h for flows .
comment: """
          Encoder that ouputs parameters for base distribution of z and context h for flows.
          """