search

anucvml / ddn

Deep Declarative Networks
MIT License
236 stars 37 forks source link

backpropagate through r and c for optimal transport layer #4

Closed sgould closed 3 years ago

sgould commented 3 years ago

Implement and test gradients for r and c in optimal transport layer as H^{-1}A^T(AH^{-1}A^T)^{-1}(AH^{-1}B - C) - H^{-1}B for appropriate B and C.