added GPU compatible Haar transform and tests

slimgroup / InvertibleNetworks.jl

A Julia framework for invertible neural networks

MIT License

152 stars 23 forks source link

Closed PetersBas closed 3 years ago

PetersBas commented 3 years ago

added GPU compatible 2D and 3D Haar transform and inverse Haar transform via lifting.
added adjoint test, invertibility test, and norm test

mloubout commented 3 years ago

Talked offline with bas, gtg, merging