Why are you taking the output of the last time step to generate the neural net?

[TharunMohandoss]

nascell-automl/train.py

Line 46 in 18e5e59

return outputs[:, -1:, :]

Won't this return the output of the last time step alone? To generate the neural network, we need output from every time step right? What am I missing here?

[jackyluo-learning]

nascell-automl/train.py

Line 46 in 18e5e59

return outputs[:, -1:, :]

Won't this return the output of the last time step alone? To generate the neural network, we need output from every time step right? What am I missing here?

Have you figure this out? I have the same confusion on this.