Я построил модель LSTM через keras (используя tf.keras.layers.LSTM). В модели LSTM существует четыре различных типа смещения. У меня также есть следующий код для извлечения вектора смещения из сохраненного файла контрольных точек:
from tensorflow.python import pywrap_tensorflow
checkpoint_path = get_checkpoint
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
tensor_name=[]
tensor_content=[]
for key in var_to_shape_map:
tensor_name.append(key)
tensor_content.append(reader.get_tensor(key))
bias=reader.get_tensor('layer_with_weights-1/bias/.ATTRIBUTES/VARIABLE_VALUE')
, тогда я получил только один единственный вектор смещений. Есть ли способ узнать, как четыре различных типа предубеждений организованы в этом списке?