텐서플로우:: ops:: RestoreV2

#include <io_ops.h>

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

요약

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

  • 이 작업은 먼저 "prefix"가 가리키는 V2 인덱스 파일을 찾고, 찾으면 해당 파일을 V2 체크포인트로 읽어들이는 것을 시도합니다.
  • 그렇지 않으면 V1 읽기 경로가 호출됩니다. 하지만 V1 읽기로 되돌아가는 기능이 더 이상 사용되지 않거나 결국 제거될 수 있으므로 이 동작에 의존하는 것은 권장되지 않습니다.

기본적으로 지정된 텐서를 전체적으로 복원합니다. 호출자가 저장된 텐서의 특정 부분만 복원하려면 "shape_and_slices"는 비어 있지 않은 문자열이어야 하며, 그에 따라 형식도 올바릅니다.

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

논거:

  • 범위: Scope 객체
  • 접두사: 단일 요소만 포함해야 합니다. V2 체크포인트의 접두사입니다.
  • tensor_names: shape {N}. 복원할 텐서의 이름입니다.
  • shape_and_slices: shape {N}. 복원할 텐서의 슬라이스 사양입니다. 빈 문자열은 분할되지 않은 텐서를 나타냅니다.
  • dtypes: shape {N}. 텐서에 대해 예상되는 dtype 목록입니다. 체크포인트에 저장된 dtype과 일치해야 합니다.

보고:

  • OutputList : shape {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

공공 기능

RestoreV2

 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