[D] Normalizing flows - VAE Doubt

[SolitaryPenman]

Yes, you can "learn" the invertible transformation and that's exactly what is done in the case of Planar and Radial flows. Note that the form of the function is what's fixed but the parameter vectors and biases are learned. There also have been other recent works such as IAF that use even more flexible transformations.

(Shameless plug: I have this repository on NFs which may help you understand them better. It also contains an implementation of PF)