pytorch LSTM圖像分類

一個問題:pytorch官方文檔對LSTM的輸入參數的格式是input of shape (seq_len, batch, input_size),可是本例中images.reshape(-1, sequence_length, input_size)的輸入格式爲batch,seq_len, input_size,是否是官文寫錯了? import torch import torch.nn as
相關文章
相關標籤/搜索