Ouput dimension in Pytorch LSTM

Viewed 21

I have about (10000, 6) time series data.

I made a sequence with 10 bundles.

The whole data consisted of (9990, 10, 6).

If I use a batch size of 20 and put it as an LSTM(batch_first=True) input, is the output (20, 10, n) correct?

I didn't understand why the value of n comes out as hidden_size in pytorch.


How is the output of an RNN unit determined?

6 inputs 1 output per unit?

If I'm right, the output should be (batch_size, hidden_size, 1)

0 Answers
Related