Thank you for this! It looks promising. I am a bit confused about implementation details. For the encoder it is pretty straightforward to add 2 (or 3) more channels to the images. But how should I do this for the decoder? The input to my decoder is (ideally) a 2 dimensional vector. How do I use the CoordDeconv on that? Thank you!