Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

tensorflow :: ops :: PulihkanV2

#include <io_ops.h>

Mengembalikan tensor dari pos pemeriksaan V2.

Ringkasan

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

  • Operasi ini pertama kali mencoba menemukan file indeks V2 yang ditunjukkan oleh "awalan", dan jika ditemukan, lanjutkan untuk membacanya sebagai pos pemeriksaan V2;
  • Jika tidak, jalur baca V1 akan dipanggil. Mengandalkan perilaku ini tidak disarankan, karena kemampuan untuk kembali membaca V1 mungkin tidak digunakan lagi dan pada akhirnya dihapus.

Secara default, memulihkan tensor bernama secara penuh. Jika pemanggil ingin memulihkan irisan tertentu dari tensor yang disimpan, "shape_and_slices" harus berupa string yang tidak kosong dan dibentuk dengan baik.

Penelepon harus memastikan semua tensor bernama memang disimpan di checkpoint.

Argumen:

  • scope: Objek Scope
  • awalan: Harus memiliki satu elemen. Awalan dari pos pemeriksaan V2.
  • tensor_names: bentuk {N}. Nama tensor yang akan dipulihkan.
  • shape_and_slices: bentuk {N}. Spesifikasi irisan tensor yang akan dipulihkan. String kosong menunjukkan bahwa mereka adalah tensor non-partisi.
  • dtypes: bentuk {N}. Daftar tipe yang diharapkan untuk tensor. Harus cocok dengan yang disimpan di pos pemeriksaan.

Pengembalian:

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

Pembuat dan Penghancur

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

PulihkanV2

 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