Thanks again for your prompt reply @ptrblck .
The current value of theta_estimated is the initial value. I didn’t initialize with zero, I initialized with that random tensor. So it is not updating.  Assigning shape_parameters_estimated to theta_estimated shouldn’t be a problem, right?