AutoregressiveNetwrok doc example

Hello,

I am not so familiar with TF so I may misunderstood something in the machinery.

Below, I just C&P the example from the doc found here

# Construct and fit model.
  x_ = tfkl.Input(shape=(2,), dtype=tf.float32)
  log_prob_ = distribution.log_prob(x_)
  model = tfk.Model(x_, log_prob_)
  model.compile(optimizer=tf.optimizers.Adam(),
                loss=lambda _, log_prob: -log_prob)

I was wandering why the loss in the model.compile does not use tf.reduce_mean ie

loss=lambda _, log_prob: -tf.reduce_mean(log_prob)

Am I wrong? Thanks

I am trying to do that and let you know. But Please Give Proper solution.

@Amin_Jabari,

We notice your response with an unusual hyperlink for the above query. Could you help us to understand the purpose? We are here to help you to resolve your problem. Thank you.