search

anucvml / ddn

Deep Declarative Networks
MIT License
237 stars 37 forks source link

Update torch.linalg.solve to the latest pytorch #21

Closed disungatullina closed 5 months ago

disungatullina commented 5 months ago

https://github.com/anucvml/ddn/blob/master/ddn/pytorch/node.py#L309

Bug, it should be X[i, ...] = torch.linalg.solve(A[i, ...], B[i, ...])