텐서플로우:: 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