PaddlePaddle

 paddle.nn / RNNCellBase


RNNCellBase

class paddle.nn. RNNCellBase ( name_scope=None, dtype='float32' ) [源代码]

循环神经网络单元基类

该OP(RNNCellBase)是一个抽象表示根据输入和隐藏状态来计算输出和新状态的基本类,最适合也最常用于循环神经网络。

get_initial_states(batch_ref,shape=None,dtype=None,init_value=0.,batch_dim_idx=0):

根据输入的形状,数据类型和值生成初始状态。

参数:
  • batch_ref (Tensor) - 一个Tensor,其形状决定了生成初始状态使用的batch_size。当batch_ref形状为d时,d[batch_dim_idx]为batch_size。

  • shape (list|tuple,可选) - 隐藏层的形状(可以是多层嵌套的),列表或元组的第一位为batch_size,默认为-1。shape为None时,使用state_shape(property)。默认为None。

  • dtype (str|list|tuple,可选) - 数据类型(可以是多层嵌套的,但嵌套结构要和shape相同。或者所有Tensor的数据类型相同时可以只输入一个dtype。)。当dtype为None且state_dtype(property)不可用时,则使用paddle默认的float类型。默认为None。

  • init_value (float,可选) -用于初始状态的浮点数值。默认为0。

  • batch_dim_idx (int,可选) - 用于指定batch_size在batch_ref的索引位置的整数值。默认为0。

返回:
  • init_state (Tensor|tuple|list) - 根据输出的数据类型,形状和嵌套层级返回的初始状态Tensor。


此页内容是否对您有帮助