【Tensorflow】RNN常用函数

最新推荐文章于 2024-09-07 22:06:23 发布

rainie1003

最新推荐文章于 2024-09-07 22:06:23 发布

阅读量516

点赞数 1

分类专栏： Tensorflow

本文链接：https://blog.youkuaiyun.com/xykimred/article/details/88780626

版权

本文介绍了TensorFlow中RNN的基本单元RNNCell，包括其功能和两个主要子类BasicRNNCell和BasicLSTMCell。此外，讨论了如何一次性执行多步RNN操作以及如何通过MultiRNNCell堆叠多层RNN。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

整理自：TensorFlow中RNN实现的正确打开方式：https://blog.youkuaiyun.com/starzhou/article/details/77848156

RNN的基本单元“RNNcell”

(output, next_state) = call(input, state)。
- 每调用一次RNNCell的call方法，就相当于在时间上“推进了一步”，这就是RNNCell的基本功能。
- 执行一次，序列时间上前进一步。
- 有两个子类：BasicRNNCell和BasicLSTMCell

cell = tf.nn.rnn_cell.BasicRNNCell(num_units=128)
print(cell.state_size) # 隐藏层的大小：128

inputs = tf.placeholder(np.float32,shape=(32,100)) # 32为batch_size
h0 = cell.zero_state(32