Не могу создать TensorDataset (Pytorch)

Всем привет! У меня имеется датафрейм "df" с двумя колонками - "X_Label" и "Y_Label", в каждом из которых содержатся тензоры. Вот пример для понимания моей ситуации:

df["X_Label"].head()
0    [[[tensor(179.), tensor(213.), tensor(321.), t...
1    [[[tensor(175.), tensor(123.), tensor(217.), t...
2    [[[tensor(135.), tensor(27.), tensor(7.), tens...
3    [[[tensor(299.), tensor(277.), tensor(265.), t...
4    [[[tensor(95.), tensor(251.), tensor(343.), te...
Name: X_Label, dtype: object

df["X_Label"][1]
tensor([[[175., 123., 217.,  ...,  45., 117., 121.],
     [ 23., 117., 121.,  ..., 209., 281., 233.],
     [209., 281., 231.,  ..., 259., 133., 167.],
     ...,
     [239., 313., 269.,  ..., 259., 165., 181.],
     [185.,  79.,  43.,  ..., 119., 265., 241.],
     [115., 355., 321.,  ..., 239., 313., 271.]],

    [[239., 313., 269.,  ..., 259., 165., 181.],
     [187.,  77.,  43.,  ..., 115., 225., 199.],
     [ 75., 279., 243.,  ..., 239., 313., 271.],
     ...,
     [275., 353., 301.,  ..., 247., 309., 265.],
     [247., 309., 265.,  ..., 267., 173., 189.],
     [255., 197., 197.,  ..., 273., 353., 301.]],

    [[275., 353., 301.,  ..., 247., 309., 265.],
     [247., 309., 265.,  ..., 267., 173., 189.],
     [251., 197., 181.,  ..., 273., 353., 301.],
     ...,
     [145., 159., 303.,  ..., 317., 173., 171.],
     [319., 165., 165.,  ..., 231., 293., 249.],
     [231., 293., 251.,  ..., 325., 321., 303.]]])

df["Y_Lable"].head()
0    tensor(0)
1    tensor(0)
2    tensor(0)
3    tensor(0)
4    tensor(0)
Name: Y_Lable, dtype: object

df["Y_Lable"][1]
tensor(0)

Я пытаюсь выполнить следующий код, чтобы поместить все данные в TensorDataset:

tdataset = torch.utils.data.TensorDataset(torch.tensor(df["X_Label"]), torch.tensor(df["Y_Lable"]))

На что мне выдает ошибку:

ValueError                                Traceback (most recent call last)

in ----> 1 tdataset = torch.utils.data.TensorDataset(torch.tensor(df["X_Label"]), torch.tensor(df["Y_Lable"]))

ValueError: only one element tensors can be converted to Python scalars

Что не так? Как поменять чтобы работало?