如何在张量流中初始化ConvLSTMCell的num_units?

时间:2018-08-07 08:13:40

标签: tensorflow neural-network conv-neural-network lstm

我想为ConvLSTMCell初始化隐藏状态的数量,但是尽管有LSTMCell初始化程序,但没有num_units参数。

LSTMCell初始化程序参数:

__init__(
    num_units,
    use_peepholes=False,
    cell_clip=None,
    initializer=None,
    num_proj=None,
    proj_clip=None,
    num_unit_shards=None,
    num_proj_shards=None,
    forget_bias=1.0,
    state_is_tuple=True,
    activation=None,
    reuse=None,
    name=None,
    dtype=None
)

ConvLSTMCell初始化程序参数:

__init__(
    conv_ndims,
    input_shape,
    output_channels,
    kernel_shape,
    use_bias=True,
    skip_connection=False,
    forget_bias=1.0,
    initializers=None,
    name='conv_lstm_cell'
)

0 个答案:

没有答案