Creating 2D Interpolation of Samples randomly generated from VAE

Hey Everyone,

I am trying to generate samples from VAE. I want to achieve similar to the below image(taken from paper), where one image smoothly transitions from one to another.

I know how to interpolate between two images: Get the latent vectors z1 and z2 for both image and interpolate between them i.e. new_z = (1-p)z1 + pz2 and decode them using VAE.decoder.

I am stuck in 2d interpolation.