tensorflow::ops::RestoreV2

#include <io_ops.h>

Mengembalikan tensor dari checkpoint V2.

Ringkasan

Untuk kompatibilitas mundur dengan format V1, Op ini saat ini juga memungkinkan pemulihan dari checkpoint V1:

  • Operasi ini pertama-tama mencoba menemukan file indeks V2 yang ditunjuk oleh "prefix", dan jika ditemukan, akan dilanjutkan dengan membacanya sebagai titik pemeriksaan V2;
  • Jika tidak, jalur baca V1 akan dipanggil. Mengandalkan perilaku ini tidak disarankan, karena kemampuan untuk kembali membaca V1 mungkin akan dihentikan dan akhirnya dihapus.

Secara default, fungsi ini mengembalikan tensor yang disebutkan secara penuh. Jika pemanggil ingin mengembalikan bagian tertentu dari tensor yang tersimpan, "shape_and_slices" harus berupa string yang tidak kosong dan sesuai formatnya.

Pihak pemanggil harus memastikan bahwa semua tensor yang disebutkan memang tersimpan di dalam checkpoint.

Argumen:

  • Cakupan: Objek Cakupan
  • Awalan: Harus memiliki satu elemen. Awalan dari pos pemeriksaan V2.
  • tensor_names: shape {N}. Nama-nama tensor yang akan dipulihkan.
  • shape_and_slices: shape {N}. Spesifikasi irisan tensor yang akan dipulihkan. String kosong menunjukkan bahwa tensor tersebut tidak dipartisi.
  • dtypes: shape {N}. Daftar tipe data yang diharapkan untuk tensor. Harus sesuai dengan yang tersimpan di checkpoint.

Mengembalikan:

  • OutputList : bentuk {N}. Tensor yang dipulihkan, yang bentuknya dibaca langsung dari checkpoint.

Konstruktor dan Destruktor

RestoreV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input prefix, :: tensorflow::Input tensor_names, :: tensorflow::Input shape_and_slices, const DataTypeSlice & dtypes)

Atribut publik

operation
tensors

Fungsi publik

operator[] (size_t index) const

Atribut publik

operasi

Operation operation

tensor

::tensorflow::OutputList tensors

Fungsi publik

RestoreV2

 RestoreV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input prefix,
  ::tensorflow::Input tensor_names,
  ::tensorflow::Input shape_and_slices,
  const DataTypeSlice & dtypes
)

operator[]

::tensorflow::Output operator[](
  size_t index
) const