tf.compat.v1.nn.rnn_cell.BasicLSTMCell

可能已经弃用,现在用的是 tf.compat.v1.nn.rnn_cell.LSTMCell 

tf.compat.v1.nn.rnn_cell.BasicLSTMCell(
    num_units,
    forget_bias=1.0,
    state_is_tuple=True,
    activation=None,
    reuse=None,
    name=None,
    dtype=None,
    **kwargs
)

tf.compat.v1.nn.rnn_cell.LSTMCell

tf.compat.v1.nn.rnn_cell.LSTMCell(
    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,
    **kwargs
)

    num_units,LSTM单元数

猜你喜欢

转载自blog.csdn.net/Fwuyi/article/details/125680217