Google I/O에 참여해 주셔서 감사합니다. 주문형 모든 세션 보기 주문형 시청

텐서플로:: 작전:: 복원V2

#include <io_ops.h>

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

요약

V1 형식과의 역호환성을 위해 이 Op는 현재 V1 체크포인트에서도 복원할 수 있습니다.

  • 이 Op는 먼저 "접두사"가 가리키는 V2 인덱스 파일을 찾으려고 시도하고 발견되면 V2 체크포인트로 계속 읽습니다.
  • 그렇지 않으면 V1 읽기 경로가 호출됩니다. V1 읽기로 폴백하는 기능이 더 이상 사용되지 않고 결국 제거될 수 있으므로 이 동작에 의존하는 것은 권장되지 않습니다.

기본적으로 명명된 텐서를 완전히 복원합니다. 호출자가 저장된 텐서의 특정 조각을 복원하려는 경우 "shape_and_slices"는 비어 있지 않은 문자열이어야 하며 그에 따라 잘 구성되어 있어야 합니다.

호출자는 모든 명명된 텐서가 실제로 체크포인트에 저장되어 있는지 확인해야 합니다.

인수:

  • 범위: 범위 개체
  • 접두사: 단일 요소가 있어야 합니다. V2 체크포인트의 접두사.
  • tensor_names: {N} 모양. 복원할 텐서의 이름입니다.
  • shape_and_slices: {N} 모양. 복원할 텐서의 슬라이스 사양입니다. 문자열은 분할되지 않은 텐서임을 나타냅니다.
  • dtypes: {N} 모양. 텐서의 예상 dtype 목록입니다. 체크포인트에 저장된 것과 일치해야 합니다.

보고:

  • OutputList : {N} 모양. 체크포인트에서 직접 모양을 읽는 복원된 텐서입니다.

생성자와 소멸자

RestoreV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input prefix, :: tensorflow::Input tensor_names, :: tensorflow::Input shape_and_slices, const DataTypeSlice & dtypes)

공개 속성

operation
tensors

공공 기능

operator[] (size_t index) const

공개 속성

작업

Operation operation

텐서

::tensorflow::OutputList tensors

공공 기능

복원V2

 RestoreV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input prefix,
  ::tensorflow::Input tensor_names,
  ::tensorflow::Input shape_and_slices,
  const DataTypeSlice & dtypes
)

운영자[]

::tensorflow::Output operator[](
  size_t index
) const