如何使用protobuf文件存储和读取Tensorflow-Hub句子嵌入？

Question

我正在尝试使用TensorFlow的自然语言处理脚本在PDF上运行语义搜索。我想缓存本地嵌入，因为它们需要很长时间才能创建。我希望在protobuf文件中执行此操作，但是我打开了其他建议。我将如何去做？

这是代码的样子。嵌入是根据我已经初始化的一系列句子创建的。

tf_sentences = tf.reshape(sentences[0:1000], [-1])

embeddings = embed(
            tf_sentences,
            signature="default",
            as_dict=True)["default"]

with tf.Session() as sess:
  sess.run(tf.global_variables_initializer())
  sess.run(tf.tables_initializer())
  x = sess.run(embeddings)