forked from huawei/mindspore2022
35 lines
1.5 KiB
ReStructuredText
35 lines
1.5 KiB
ReStructuredText
mindspore.nn.RNNCell
|
||
=====================
|
||
|
||
.. py:class:: mindspore.nn.RNNCell(input_size: int, hidden_size: int, has_bias: bool = True, nonlinearity: str = 'tanh')
|
||
|
||
循环神经网络单元，激活函数是tanh或relu。
|
||
|
||
.. math::
|
||
h_t = \tanh(W_{ih} x_t + b_{ih} + W_{hh} h_{(t-1)} + b_{hh})
|
||
|
||
其中 :math:`h_t` 是在 `t` 时刻的隐藏状态， :math:`x_t` 是在 `t` 时刻的输入， :math:`h_{(t-1)}` 是在 :math:`t-1` 时刻的隐藏状态，或初始隐藏状态。
|
||
|
||
如果 `nonlinearity` 是'relu'，则使用'relu'而不是'tanh'。
|
||
|
||
**参数：**
|
||
|
||
- **input_size** (int) - 输入层输入的特征向量维度。
|
||
- **hidden_size** (int) - 隐藏层输出的特征向量维度。
|
||
- **has_bias** (bool) - Cell是否有偏置项 `b_ih` 和 `b_hh` 。默认值：True。
|
||
- **nonlinearity** (str) - 用于选择非线性激活函数。取值可以是'tanh'或'relu'。默认值：'tanh'。
|
||
|
||
**输入：**
|
||
|
||
- **x** (Tensor) - 输入Tensor，其shape为 :math:`(batch\_size, input\_size)` 。
|
||
- **hx** (Tensor) - 输入Tensor，其数据类型为mindspore.float32及shape为 :math:`(batch\_size, hidden\_size)` 。 `hx` 的数据类型与 `x` 相同。
|
||
|
||
**输出：**
|
||
|
||
- **hx'** (Tensor) - shape为 :math:`(batch\_size, hidden\_size)` 的Tensor。
|
||
|
||
**异常：**
|
||
|
||
- **TypeError** - `input_size` 或 `hidden_size` 不是int或不大于0。
|
||
- **TypeError** - `has_bias` 不是bool。
|
||
- **ValueError** - `nonlinearity` 不在['tanh', 'relu']中。 |