Tensorflow中的可变范围问题

当您创建BasicLSTMCell()时,它会创建所有必需的权重和偏差,以实现引擎盖下的LSTM单元.所有这些变量都自动分配名称.如果您在同一范围内多次调用该函数,则会得到错误.由于您的问题似乎表明您要创建两个单独的LSTM单元格,因此您不希望重用这些变量,但您确实希望在不同的范围内创建它们.您可以通过两种不同的方式执行此操作(我实际上没有尝试运行此代码,但它应该可以工作).您可以在一个独特的范围内调用您的函数

def biLSTM(data, n_steps):
    ... blah ...

with tf.variable_scope('LSTM1'):
    outputs, hidden = biLSTM(data, steps)

with tf.variable_scope('LSTM2'):
    outputs, hidden = biLSTM(data, steps)

或者您可以将唯一的范围名称传递给函数并使用范围内部

def biLSTM(data, n_steps, layer_name):
    ... blah...
    with tf.variable_scope(layer_name) as scope:
        lstm_fw_cell = tf.nn.rnn_cell.BasicLSTMCell(n_hidden, forget_bias=1.0)
        lstm_bw_cell = tf.nn.rnn_cell.BasicLSTMCell(n_hidden, forget_bias=1.0)
        outputs, _, _ = tf.nn.bidirectional_rnn(lstm_fw_cell, lstm_bw_cell, data, dtype=tf.float32)
    return outputs, n_hidden

l1 = biLSTM(data, steps, 'layer1')
l2 = biLSTM(data, steps, 'layer2')

这取决于您的编码敏感度选择方法,它们在功能上几乎相同.

当您创建BasicLSTMCell()时,它会创建所有必需的权重和偏差,以实现引擎盖下的LSTM单元.所有这些变量都自动分配名称.如果您在同一范围内多次调用该函数,则会得到错误.由于您的问题似乎表明您要创建两个单独的LSTM单元格,因此您不希望重用这些变量,但您确实希望在不同的范围内创建它们.您可以通过两种不同的方式执行此操作(我实际上没有尝试运行此代码,但它应该可以工作).您可以在一个独特的范围内调用您的函数

def biLSTM(data, n_steps):
    ... blah ...

with tf.variable_scope('LSTM1'):
    outputs, hidden = biLSTM(data, steps)

with tf.variable_scope('LSTM2'):
    outputs, hidden = biLSTM(data, steps)

或者您可以将唯一的范围名称传递给函数并使用范围内部

def biLSTM(data, n_steps, layer_name):
    ... blah...
    with tf.variable_scope(layer_name) as scope:
        lstm_fw_cell = tf.nn.rnn_cell.BasicLSTMCell(n_hidden, forget_bias=1.0)
        lstm_bw_cell = tf.nn.rnn_cell.BasicLSTMCell(n_hidden, forget_bias=1.0)
        outputs, _, _ = tf.nn.bidirectional_rnn(lstm_fw_cell, lstm_bw_cell, data, dtype=tf.float32)
    return outputs, n_hidden

l1 = biLSTM(data, steps, 'layer1')
l2 = biLSTM(data, steps, 'layer2')

这取决于您的编码敏感度选择方法,它们在功能上几乎相同.