original_docstring: 
          Encoder that ouputs parameters for base distribution of z and flow parameters.
          
docstring: Encoder that ouputs parameters for base distribution of z and flow parameters.
docstring_tokens: Encoder that ouputs parameters for base distribution of z and flow parameters .
code: def encode(self, x):
          """
          Encoder that ouputs parameters for base distribution of z and flow parameters.
          """
  
          batch_size = x.size(0)
  
          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 amortized u an w for all flows
          u = self.amor_u(h).view(batch_size, self.num_flows, self.z_size, 1)
          w = self.amor_w(h).view(batch_size, self.num_flows, 1, self.z_size)
          b = self.amor_b(h).view(batch_size, self.num_flows, 1, 1)
  
          return mean_z, var_z, u, w, b
code_tokens: def encode ( self , x ) : batch_size = x . size ( 0 ) 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 ) u = self . amor_u ( h ) . view ( batch_size , self . num_flows , self . z_size , 1 ) w = self . amor_w ( h ) . view ( batch_size , self . num_flows , 1 , self . z_size ) b = self . amor_b ( h ) . view ( batch_size , self . num_flows , 1 , 1 ) return mean_z , var_z , u , w , b
short_docstring: Encoder that ouputs parameters for base distribution of z and flow parameters.
short_docstring_tokens: Encoder that ouputs parameters for base distribution of z and flow parameters .
comment: """
          Encoder that ouputs parameters for base distribution of z and flow parameters.
          """$SEP$# return amortized u an w for all flows