RNNCell

    RNNCell是抽象的基类,代表将输入和状态映射到输出和新状态的计算,主要用于RNN。

    call ( inputs, states, \*kwargs* )

    每个cell都必须实现此接口,将(输入和状态)映射到(输出和新状态)。为了更灵活,输入和状态都可以是单个tensor变量或嵌套结构的tensor变量(列表 | 元组 | 命名元组 | 字典)。

    参数:

    • inputs - 输入,为单个tensor变量或tensor变量组成的嵌套结构。

    • states - 状态,单个tensor变量或tensor变量组成的嵌套结构。

    返回类型:tuple

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

    该接口根据提供的形状,数据类型和初始值来初始化状态。

    参数:

    • batch_ref - 单个tensor变量或tensor组成的嵌套结构。 tensor的第一维将用作初始化状态的batch大小。

    • shape - 单个形状或形状组成的嵌套结构,单个形状是整数的列表或元组。 如果形状的第一维不是batch大小,则自动插入-1作为batch大小。 如果该项为None,将使用属性 state_shape。默认值为None。

    • batch_dim_idx - 用于指示 batch_ref 中batch所在维度的int值,默认值为0。

    返回:和shape具有相同结构的tensor变量,代表初始状态。

    返回类型:Variable

    state_shape ( )

    抽象方法(属性),该接口用于初始化cell的状态。 单个形状或由形状组成的嵌套结构,单个形状可以是整数的列表或元组(如果形状的第一维不是batch大小,则自动插入-1作为batch大小)。 当没有使用 get_initial_states 初始化状态或 get_initial_states 没有提供 参数的时候,不用实现该方法。

    state_dtype ( )

    抽象方法(属性),该接口用于初始化cell的状态。 单个数据类型或由数据类型组成的嵌套结构,该结构必须与 shape 的结构相同,例外是当状态中的所有tensor都具有相同的数据类型,这时可以使用单个数据类型。 当没有使用 get_initial_states 初始化状态或 get_initial_states 没有提供 参数的时候,不用实现该方法。