GaussianVisLayer handles arbitrary axes order