텐서플로우 C++ 세션 API 레퍼런스 문서

0.5 버전의 텐서플로우의 퍼블릭 C++ API는 오직 그래프를 실행하는 API만을 포함합니다. C++ 로 부터 그래프 실행을 하는 것은 다음과 같습니다.

  1. Python API를 이용해서 산출 그래프를 빌드합니다.
  2. 그래프를 파일에 쓰기위해 tf.train.write_graph()를 이용합니다.

  3. C++ 세션 API를 이용해 그래프를 읽어옵니다. 예를 들면:

    // Reads a model graph definition from disk, and creates a session object you
// can use to run it.
Status LoadGraph(string graph_file_name, Session** session) {
 GraphDef graph_def;
 TF_RETURN_IF_ERROR(
     ReadBinaryProto(Env::Default(), graph_file_name, &graph_def));
 TF_RETURN_IF_ERROR(NewSession(SessionOptions(), session));
 TF_RETURN_IF_ERROR((*session)->Create(graph_def));
 return Status::OK();
}

  4. session->Run()을 호출하여 그래프를 보여줍니다.

Env

Session

Status

Tensor

Thread

results matching ""

    No results matching ""