Method encode

examples/models/contrib/stdit/vae.py:576–584

(self, x)

        return latent_size

    def encode(self, x):
        time_padding = x.shape[2] % self.time_downsample_factor
        if time_padding != 0:
            time_padding = self.time_downsample_factor - time_padding
        x = pad_at_dim(x, (time_padding, 0), dim=2)
        encoded_feature = self.encoder(x)
        moments = self.quant_conv(encoded_feature).to(x.dtype)
        posterior = DiagonalGaussianDistribution(moments)
        return posterior

    def decode(self, z, num_frames=None):
        time_padding = num_frames % self.time_downsample_factor