Modified based on code in https://github.com/gr8joo/MVTCAE.