Reshape tensor from 2d to 3d.