tensorflow 導入gfile模型文件

with tf.gfile.GFile(os.path.join(self.model_dir, 'ner_model.pb'), 'rb') as f:
      graph_def = self.tf.GraphDef()
      graph_def.ParseFromString(f.read())
      input_map = {"input_ids:0": self.input_ids,
                             'input_mask:0': self.input_mask}
       # 這就是我們要獲取的op
      self.pred_ids = self.tf.import_graph_def(graph_def,
                                                         name='',
                                                         input_map=input_map,
                                                         return_elements=['pred_ids:0'])[0]
      graph = self.pred_ids.graph