tensorflow:: ops:: RestoreV2

#include <io_ops.h>

V2チェックポイントからテンソルを復元します。

まとめ

V1フォーマットとの下位互換性を確保するため、この操作では現在、V1チェックポイントからの復元も可能になっています。

  • この操作では、まず「prefix」で指定されたV2インデックスファイルを検索し、見つかった場合はそれをV2チェックポイントとして読み込みます。
  • そうでない場合は、V1読み取りパスが呼び出されます。この動作に依存することは推奨されません。V1読み取りへのフォールバック機能は非推奨となり、最終的には削除される可能性があるためです。

デフォルトでは、指定されたテンソル全体が復元されます。呼び出し元が保存されたテンソルの特定のスライスを復元したい場合は、「shape_and_slices」は空でない文字列であり、かつ適切な形式である必要があります。

呼び出し元は、指定されたすべてのテンソルがチェックポイントに正しく格納されていることを確認する必要があります。

議論:

  • スコープ:スコープオブジェクト
  • 接頭辞:要素は1つだけである必要があります。V2チェックポイントの接頭辞です。
  • tensor_names: shape {N}。復元するテンソルの名前。
  • shape_and_slices: 形状 {N}。復元するテンソルのスライス仕様。空の文字列は、それらが分割されていないテンソルであることを示します。
  • dtypes: shape {N}。テンソルに期待される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

公的機能

リストア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