What are the output shape of LSTM with "return_sequences" equal to "True" or "False"?

Question

What are the output shape of LSTM with Keras implementation when "return_sequences" equals to "True" or "False" ?

score 2 · Accepted Answer · answered Mar 02 '19 at 10:41

2

Assuming your output state size is h. You will generate nxh for n data points if return_sequences is false. On the other hand you would be having nxtxh dimensional array as output if return_sequences is set to true. Here t is the number of timestamps in each data point.

answered Mar 02 '19 at 10:41

Kiritee Gak

1,799
1
11
25

So choosing return_sequences = True/False doesn't change the 3D array output shape? – user3486308 Mar 02 '19 at 11:08
It does. In false case you got last hidden state out. But in true case every internal hidden state gets released out. – Kiritee Gak Mar 02 '19 at 11:10
Aha, you meant one dimension vector by h. – user3486308 Mar 02 '19 at 13:37
Yes. That is right. – Kiritee Gak Mar 02 '19 at 14:00

What are the output shape of LSTM with "return_sequences" equal to "True" or "False"?

1 Answers1