Hi all,
I have a encoder network with 3 layer and hidden size of 40 and an decoder network with 5 layer and hidden size of 30. How should I connect the output of encoder network to the input of decoder network?
As I know nn.Linear may works, but nn.Linear input is (N, *, in_feature) while encoder output is (3, N, hidden_size) and I couldn’t figure out how should I convert (3, N, hidden_size) vector to suitable input for nn.Linear module. I used view method, but it changes orders.