def _autoregressive_flow_kwargs(self):
   return {
       "shift_and_log_scale_fn": masked_autoregressive_default_template(
           hidden_layers=[2], shift_only=False),
       "is_constant_jacobian": False,
       "unroll_loop": True,
   }
예제 #2
0
 def _autoregressive_flow_kwargs(self):
     return {
         "shift_and_log_scale_fn":
         masked_autoregressive_default_template(hidden_layers=[2],
                                                shift_only=False),
         "is_constant_jacobian":
         False,
     }