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

#include <io_ops.h>

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

요약

V1 형식과의 이전 버전 호환성을 위해 이 작업은 현재 V1 체크포인트에서도 복원을 허용합니다.

  • 이 작업은 먼저 "접두사"가 가리키는 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