theta_estimated is assigned to shape_parameters_estimated, which was initialized with torch.zeros. In your current output theta_estimated shows values which are different from zeros, so I guess that it was already updated?