Will it be possible to support multivariate normal distributions just like pytorch does, where user can provide custom covariance matrix in the future?
I want to implement a VAE where the inference model produce a mean and a first-order autoregressive covariance matrix, and the posterior distribution is multivariate.
Will it be possible to support multivariate normal distributions just like pytorch does, where user can provide custom covariance matrix in the future? I want to implement a VAE where the inference model produce a mean and a first-order autoregressive covariance matrix, and the posterior distribution is multivariate.