代码之家 › 专栏 › 技术社区 › MatthewScarpino

TensorFlow的RNN单元和信元

rnn neural-network tensorflow

0

MatthewScarpino · 技术社区 · 8 年前

许多RNN类的构造函数( BasicRNNCell , LSTMCell num_units . 这将设置单元中的单元数。

我认为这确定了RNN应该按顺序处理的元素的数量。因此,如果您希望RNN处理长度为N的序列,则每个单元有N个单元。这是正确的吗?RNN单元到底是什么?

2 回复 | 直到 8 年前

1

2

Lorenzo D'Arsiè 8 年前

不,这是不对的。

num_units 指单元格可以表示的特征数。在每个时间步骤中,您都会给出一个特定大小的输入(您称之为“RNN应按顺序处理的元素数”)。这就像神经网络的第0层。然后将该输入处理为一个隐藏层,大小为 num_单位 . 这也是单元格输出的大小。

你所说的N,是由你的输入张量的大小决定的。是模型的超参数。它越大,模型的自由度就越大(描述特征越多)。

2

Pegasus 8 年前

这里num_units是指LSTM(或rnn)单元中的单元数。