Extend to multi-channel data

[ziatdinovmax]

Currently, all the VED/VAE models work only with a single channel data (e.g. grayscale images but not color images). The extension to multiple channels should be pretty straightforward if there is a use case.