pytorch 中的 torch.nn.RNN 的參數

1、定義RNN的網絡結構的參數(類似於CNN中定義 in_channel,out_channel,kernel_size等等)          input_size   輸入x的特徵大小(以mnist圖像爲例,特徵大小爲28*28 = 784)          hidden_size   隱藏層h的特徵大小          num_layers    循環層的數量(RNN中重複的部分)  
相關文章
相關標籤/搜索