此页面由 Cloud Translation API 翻译。
Switch to English

tf.compat.v1.lite.experimental.nn.TfLiteRNNCell

GitHub上查看源代码

最基本的RNN细胞。

这仅用于TfLite,它提供提示和这也使得在所需的tflite OPS的变量。

num_units INT,在RNN电池单电池单元的数量。
activation 非线性使用。默认: tanh 。它也可能是字符串,它是Keras激活函数名之内。
reuse (可选)的Python布尔描述是否重用变量在现有范围。引发一个错误,如果没有True和现有的范围已经给定的变量。
name 字符串,该层的名称。具有相同名称的图层会分享的权重,但为了避免我们需要在这样的情况下重用=真正的错误。
dtype 该层的默认D型细胞(的默认None装置使用第一个输入的类型)。在需要时build之前被调用call
**kwargs 字典,对于公共层的属性的关键字命名属性,像trainable从get_config的CONFIGS构建单元时等()。

ValueError 如果现有的范围已经给定的变量。

graph 推荐的函数

output_size 整数或TensorShape:由该细胞产生的输出大小。
scope_name

state_size 尺寸由该小区中使用的状态(多个)的(一个或多个)。

它可以由一个整数,a TensorShape或整数或TensorShapes的元组来表示。

方法

get_initial_state

查看源代码

zero_state

查看源代码

返回零填充状态张量(一个或多个)。

ARGS
batch_size 整数,浮点,或单元张量表示批量大小。
dtype 数据类型要用于状态。

返回
如果state_size是int或TensorShape,则返回值是一个ND形状的张量[batch_size, state_size]用零填充。

如果state_size是嵌套列表或元组,则返回值是的嵌套列表或元组(具有相同的结构) 2-D与形状张量[batch_size, s]为在各S- state_size