przepływ tensorowy:: ops:: Przywrócić
#include <io_ops.h>
Przywraca tensor z plików punktów kontrolnych.
Streszczenie
Odczytuje tensor zapisany w jednym lub kilku plikach. Jeśli istnieje kilka plików (na przykład dlatego, że tensor został zapisany jako wycinki), file_pattern
może zawierać symbole wieloznaczne ( *
i ?
) tylko w części zawierającej nazwę pliku, a nie w części katalogu.
Jeśli file_pattern
pasuje do kilku plików, preferred_shard
można użyć do wskazania, w którym pliku prawdopodobnie znajdzie się żądany tensor. Ta operacja najpierw otworzy plik o indeksie preferred_shard
na liście pasujących plików i spróbuje przywrócić tensory z tego pliku. Tylko jeśli w pierwszym pliku nie zostaną znalezione niektóre tensory lub wycinki tensora, Op otwiera wszystkie pliki. Ustawienie preferred_shard
na wartość zgodną z wartością przekazaną jako dane wejściowe shard
pasującego Save
zapisu może przyspieszyć przywracanie . Ten atrybut wpływa tylko na wydajność, a nie na poprawność. Wartość domyślna -1 oznacza, że pliki są przetwarzane w kolejności.
Zobacz także RestoreSlice
.
Argumenty:
- zakres: Obiekt Scope
- file_pattern: Musi zawierać pojedynczy element. Wzór plików, z których odczytujemy tensor.
- tensor_name: Musi zawierać pojedynczy element. Nazwa tensora, który ma zostać przywrócony.
- dt: Typ tensora, który ma zostać przywrócony.
Opcjonalne atrybuty (patrz Attrs
):
- preferowany_shard: Indeks pliku, który ma zostać otwarty jako pierwszy, jeśli wiele plików pasuje do
file_pattern
.
Zwroty:
-
Output
: przywrócony tensor.
Konstruktory i destruktory | |
---|---|
Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt) | |
Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs) |
Atrybuty publiczne | |
---|---|
operation | |
tensor |
Funkcje publiczne | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Publiczne funkcje statyczne | |
---|---|
PreferredShard (int64 x) |
Struktury | |
---|---|
tensorflow:: ops:: Przywróć:: Atrybuty | Opcjonalne moduły ustawiające atrybuty dla Restore . |
Atrybuty publiczne
działanie
Operation operation
napinacz
::tensorflow::Output tensor
Funkcje publiczne
Przywrócić
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt )
Przywrócić
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs )
węzeł
::tensorflow::Node * node() const
operator::tensorflow::Wejście
operator::tensorflow::Input() const
operator::tensorflow::Wyjście
operator::tensorflow::Output() const
Publiczne funkcje statyczne
Preferowany fragment
Attrs PreferredShard( int64 x )