edited GaussianVisLayer to deal with axes order