Error : y.register_hook in cvx_utils.py file

[ravinalawade13]

Original line was : y.register_hook(lambda grad,b=batch : torch.solve(grad[:,None], J[b].transpose(0,1))[0][:,0])

I changed to torch.linalg.solve, but still got error as
RuntimeError: linalg.solve: A must be batches of square matrices, but they are 65 by 1 matrices

I changed the code to the
y.register_hook(lambda grad,b=batch : torch.linalg.solve(J[b].transpose(0,1),grad[:,None])[:,0])
By this change, code works but I am not sure if operations are right ?

[kdhht2334]

Sorry for not being able to take a closer look.

In fact, I also referred to a lot of the tutorials below. I hope it helps you too.

https://implicit-layers-tutorial.org/