텐서플로우:: 작전:: 복원하다

#include <io_ops.h>

체크포인트 파일에서 텐서를 복원합니다.

요약

하나 또는 여러 파일에 저장된 텐서를 읽습니다. 여러 개의 파일이 있는 경우(예를 들어 텐서가 슬라이스로 저장되었기 때문에) file_pattern 에는 디렉토리 부분이 아닌 파일 이름 부분에만 와일드카드 기호( *? )가 포함될 수 있습니다.

file_pattern 여러 파일과 일치하는 경우 preferred_shard 사용하여 요청된 텐서를 찾을 가능성이 있는 파일을 힌트할 수 있습니다. 이 작업은 먼저 일치하는 파일 목록의 preferred_shard 인덱스에서 파일을 열고 해당 파일에서 텐서를 복원하려고 시도합니다. 일부 텐서 또는 텐서 슬라이스가 첫 번째 파일에서 발견되지 않는 경우에만 Op는 모든 파일을 엽니다. 일치하는 Save 작업의 shard 입력으로 전달된 값과 일치하도록 preferred_shard 설정하면 복원 속도가 빨라질 수 있습니다. 이 속성은 성능에만 영향을 미치며 정확성에는 영향을 미치지 않습니다. 기본값 -1은 파일이 순서대로 처리됨을 의미합니다.

RestoreSlice 도 참조하세요.

인수:

  • 범위: 범위 개체
  • file_pattern: 단일 요소가 있어야 합니다. 텐서를 읽는 파일의 패턴입니다.
  • tensor_name: 단일 요소가 있어야 합니다. 복원할 텐서의 이름입니다.
  • dt: 복원할 텐서의 유형입니다.

선택적 속성( Attrs 참조):

  • Preferred_shard: 여러 파일이 file_pattern 일치하는 경우 먼저 열 파일의 인덱스입니다.

보고:

생성자와 소멸자

Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt)
Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs)

공개 속성

operation
tensor

공공 기능

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

공개 정적 함수

PreferredShard (int64 x)

구조체

텐서플로우:: ops:: 복원:: 속성

복원을 위한 선택적 속성 설정자입니다.

공개 속성

작업

Operation operation

텐서

::tensorflow::Output tensor

공공 기능

복원하다

 Restore(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input file_pattern,
  ::tensorflow::Input tensor_name,
  DataType dt
)

복원하다

 Restore(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input file_pattern,
  ::tensorflow::Input tensor_name,
  DataType dt,
  const Restore::Attrs & attrs
)

마디

::tensorflow::Node * node() const 

연산자::텐서플로우::입력

 operator::tensorflow::Input() const 

연산자::텐서플로우::출력

 operator::tensorflow::Output() const 

공개 정적 함수

선호샤드

Attrs PreferredShard(
  int64 x
)