目前，我已經(jīng)完成了 TensorFlow 的功能神經(jīng)元模型，我想將預(yù)測輸出到 CSV 文件。然而，當(dāng)我得到預(yù)測時(shí)，它是一個(gè)具有 8 類輸出概率的張量，428 行。        0           1           2           3           4          5            6          70   0.001258    0.971390    0.000170    0.005542    0.000077    0.002042    0.001015    0.0185051   0.000663    0.989145    0.000010    0.000393    0.000642    0.000798    0.003435    0.0049142   0.001139    0.939257    0.000232    0.017595    0.000112    0.000762    0.000535    0.0403683   0.002857    0.970207    0.000041    0.000680    0.003005    0.000273    0.012651    0.0102864   0.000706    0.992306    0.000017    0.000533    0.000908    0.000095    0.000808    0.004627..428如何獲取每個(gè)類別編號(hào)的最高概率作為輸出？例如，第一個(gè)數(shù)據(jù)點(diǎn) 0，很明顯看到類別 1 的概率最高。