@jpeg729,
The thing is that hidden and cell state outputs from lstm are only of the recent timestamp.
you will be summing only the last hidden state.
I’m interested in summing these at each timestamp (many to many problem).
That’s why I tried to implement it with a for loop over the time dimention