anucvml / ddn

Deep Declarative Networks
MIT License
237 stars 37 forks source link

Update eigen_decomposition.py #18

Closed csyanbin closed 1 year ago

csyanbin commented 1 year ago

The inplace torch.where(..., out=L) in line 33 caused backward error.
Fixed.