Getting mean and covariance matrix for multivariate normal from keras model

Question

I have a dataset that has 6 input features and 5 output features. I want to use a keras sequential model to estimate the mean vector and covariance matrix from any row of in