tensorflow :: ops :: 복원

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

요약

하나 또는 여러 파일에 저장된 텐서를 읽습니다. 여러 파일이있는 경우 (예 : 텐서가 슬라이스로 저장 되었기 때문에) 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 과 일치하는 경우 먼저 열 파일 색인.

보고:

• Output : 복원 된 텐서.

공용 속성

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
)