# 從ckpt文件讀入模型參數(shù)
reader = tf.train.NewCheckpointReader(model_path)
# 獲取變量列表
var_dict = reader.get_variable_to_dtype_map()
# 從列表里面選擇想要保存的variable
tensor = reader.get_tensor('bert/embeddings/word_embeddings')
tensor.shape # (21128, 128)
然后使用np.savetxt
之類的方法存到文件里除抛。