# 从ckpt文件读入模型参数
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
之类的方法存到文件里。