increase reshape layer coverage
MemoryDataLayer can handle non-4D tensor