深度學(xué)習(xí) Kaggle新賽 有毒評論識別大賽指導(dǎo)班(NLP·文本分類)
2023-01-15 14:05 作者:bili_1559000721 | 我要投稿
維 Tensor(矩陣)
在 pytorch 中,2 維矩陣主要用于表示圖片,在計(jì)算機(jī)中,我們是用 2 維矩陣表示圖片每一個(gè)像素位置,然后在通過一個(gè)通道維度表示像素的顏色和透明度的信息。
在 mnist 數(shù)據(jù)集為例來說明 2 維矩陣在神經(jīng)網(wǎng)絡(luò)中的用途,輸入圖片為?
,然后將 2 維圖圖像展平為 1 維 784 維,然后如果一次輸入多長圖片,就是
?表示每一個(gè)批次為 32
a = torch.rand(2,3)
標(biāo)簽: