Add capability for single non-unity layer dimenson