Convert 2d tensors to 3d for Golriz data
Added wrapper to make parameter lambda in fit