Add StaticRNN.output code example (#18251)

refine StaticRNN api doc
test=develop
test=document_preview
revert-18229-add_multi_gpu_install_check
zhaoyuchen2018 6 years ago committed by GitHub
parent 2f0d68261c
commit 32c95f1752
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -307,6 +307,7 @@ class StaticRNN(object):
hidden = fluid.layers.fc(input=[word, prev], size=hidden_size, act='relu') hidden = fluid.layers.fc(input=[word, prev], size=hidden_size, act='relu')
rnn.update_memory(prev, hidden) # set prev to hidden rnn.update_memory(prev, hidden) # set prev to hidden
rnn.step_output(hidden) rnn.step_output(hidden)
rnn.output(word)
result = rnn() result = rnn()

Loading…
Cancel
Save