def split_rnn_outputs(model, rnn_outputs):
"""
Split the output of dynamic_rnn into the actual RNN outputs and the state update gate
"""
if using_skip_rnn(model):
return rnn_outputs.h, rnn_outputs.state_gate
else:
return rnn_outputs, tf.no_op()
graph_definition.py 文件源码
python
阅读 21
收藏 0
点赞 0
评论 0
评论列表
文章目录