rnn = nn.LSTM(10,20,2) -> (input_size,hidden_size,num_layers) h0 = torch.randn(2,3,20) ->(num_layers* 1,batch_size,hidden_size)
確定! 回上一頁