Я читаю CSV-файл в pandas dataframe.
train_data = pd.read_csv('mnist_test.csv');
Пример данных
label pixel1 pixel2 pixel3 ... pixel781 pixel782 pixel783 pixel784
0 6 149 149 150 ... 106 112 120 107
1 5 126 128 131 ... 184 184 182 180
2 10 85 88 92 ... 226 225 224 222
3 0 203 205 207 ... 230 240 253 255
4 3 188 191 193 ... 49 46 46 53
как я могу преобразовать этот массив данных в набор данных тензор потока.