Hi,
Thank you. It’s working without any error, but the model can not be trained, i.e., the loss doesn’t decrease.   In your code snippet, how do you involve the input x? I have this same question about your second solution as well, i.e, how can we make the chain between the input and our trainable gamma in this solution?