Function to_tensor_2d 

pub fn to_tensor_2d(data: &TensorData, device: &Device) -> Result<Tensor>