How to get the last output?

Question

How to get the last output?

qtcv opened this issue 8 years ago · comments

In dynamic_rnn.ipynb, you can get the last ouput by result[0]["outputs"][1,6]
but in most cases, the last output index could be variable, How to get the last outputs in a batch?

Denny Britz · Answer 1 · Mon Dec 26 2016 00:59:31 GMT+0800 (China Standard Time)

You can index it by example length. result[0]["outputs"][1,example_length[1]]

Alternatively you can use the last state that is returned, in some cases state and output are the same (i.e. GRU), and in some cases (LSTM) the output is a subset of the state, e.g. for an LSTM:output = state.m

qtcv · Answer 2 · Mon Dec 26 2016 08:21:00 GMT+0800 (China Standard Time)

Thanks, there is another solution: https://danijar.com/variable-sequence-lengths-in-tensorflow/

Denny Britz · Answer 3 · Mon Dec 26 2016 10:58:25 GMT+0800 (China Standard Time)

I think you are confusing two different things. What you quote about is in Python after you fetched values from the Tensoflow graph. What you link to is (a rather bad way) of doing it in-graph. Anyway, closing this for now.