mindspore.nn.GRU

class mindspore.nn.GRU(*args, **kwargs)[源代码]

GRU(Gate Recurrent Unit)称为门控循环单元网络,是循环神经网络(Recurrent Neural Network, RNN)的一种。根据输出序列和给定的初始状态计算输出序列和最终状态。

GRU网络模型中有两个门,一个是更新门,另一个是重置门。将两个连续的时间节点表示为 \(t-1\)\(t\)。给定一个在时刻 \(t\) 的输入 \(x_t\) ,一个隐藏状态 \(h_{t-1}\) 。在时刻 \(t\) 的更新门和重置门使用门控机制计算。更新门 \(z_t\) 用于控制前一时刻的状态信息被带入到当前状态中的程度。重置门 \(r_t\) 控制前一状态有多少信息被写入到当前候选集 \(n_t\) 上。完整的公式如下。

\[\begin{split}\begin{array}{ll} r_t = \sigma(W_{ir} x_t + b_{ir} + W_{hr} h_{(t-1)} + b_{hr}) \\ z_t = \sigma(W_{iz} x_t + b_{iz} + W_{hz} h_{(t-1)} + b_{hz}) \\ n_t = \tanh(W_{in} x_t + b_{in} + r_t * (W_{hn} h_{(t-1)}+ b_{hn})) \\ h_t = (1 - z_t) * n_t + z_t * h_{(t-1)} \end{array}\end{split}\]

其中 \(\sigma\) 是sigmoid激活函数, \(*\) 是乘积。 \(W,b\) 是公式中输出和输入之间的可学习权重。例如, \(W_{ir}, b_{ir}\) 是用于将输入 \(x\) 转换为 \(r\) 的权重和偏置。详见论文 Learning Phrase Representations using RNN Encoder–Decoder for Statistical Machine Translation

Note

当GRU运行在Ascend上时,hidden size仅支持16的倍数。

参数:

  • input_size (int) - 输入的大小。

  • hidden_size (int) - 隐藏状态大小。

  • num_layers (int) - 网络层数。默认值:1。

  • has_bias (bool) - cell是否有偏置项 b_{in}b_{hn} 。默认值:True。

  • batch_first (bool) - 指定输入 x 的第一个维度是否为batch_size。默认值:False。

  • dropout (float) - 指的是除第一层外每层输入时的Dropout概率。默认值:0.0。Dropout的范围为[0.0, 1.0)。

  • bidirectional (bool) - 是否为双向GRU。如果bidirectional=True,则num_directions=2,为双向GRU。否则为1,单向GRU。默认值:False。

输入:

  • x (Tensor) - 数据类型为mindspore.float32、shape为(seq_len, batch_size, input_size)或(batch_size, seq_len, input_size)的tensor。

  • hx (Tensor) - 数据类型为mindspore.float32、shape为(num_directions * num_layers , batch_size, hidden_size )的tensor。 hx 的数据类型必须与 x 相同。

  • seq_length (Tensor) - 输入batch中每个序列的长度。shape为 (batch_size) 的Tensor。默认值:None。此输入指示填充前的真实序列长度,避免填充元素被用于计算隐藏状态而影响最终输出。当 x 含填充元素时,建议使用此输入。

输出:

Tuple,包含(output, h_n)的tuple。

  • output (Tensor) - shape为(seq_len, batch_size, num_directions * hidden_size)或(batch_size, seq_len, num_directions * hidden_size)的Tensor。

  • hx_n (Tensor) - shape为(num_directions * num_layers, batch_size, hidden_size)的Tensor。

异常:

  • TypeError - input_sizehidden_sizenum_layers 不是整数。

  • TypeError - has_biasbatch_firstbibound 不是bool。

  • TypeError - dropout 既不是浮点数也不是整数。

  • ValueError - dropout 不在[0.0, 1.0)范围内。

支持平台:

Ascend GPU CPU

样例:

>>> net = nn.GRU(10, 16, 2, has_bias=True, batch_first=True, bidirectional=False)
>>> x = Tensor(np.ones([3, 5, 10]).astype(np.float32))
>>> h0 = Tensor(np.ones([1 * 2, 3, 16]).astype(np.float32))
>>> output, hn = net(x, h0)
>>> print(output.shape)
(3, 5, 16)