1d Tensors Expected But Got 2d And 2d Tensors