PyTorch快速入門教程六(使用LSTM作圖片分類)

爲何80%的碼農都作不了架構師?>>>   網絡 對於LSTM,咱們要處理的數據是一個序列數據,對於圖片而言,咱們如何將其轉換成序列數據呢?圖片的大小是28x28,因此咱們能夠將其當作長度爲28的序列,序列中的每一個數據的維度是28,這樣咱們就能夠將其變成一個序列數據了。架構 model class Rnn(nn.Module): def __init__(self, in_dim, hi
相關文章
相關標籤/搜索