텐서플로우:: 작전:: 복원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