テンソルフロー::作戦::復元V2

#include <io_ops.h>

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

まとめ

V1 形式との下位互換性のために、この Op では現在、V1 チェックポイントからの復元も許可されています。

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

デフォルトでは、名前付きテンソルを完全に復元します。呼び出し元が保存されたテンソルの特定のスライスを復元したい場合、「shape_and_slices」は空ではなく、それに応じて整形式である必要があります。

呼び出し元は、すべての名前付きテンソルが実際にチェックポイントに格納されていることを確認する必要があります。

引数:

  • スコープ:スコープオブジェクト
  • prefix: 単一の要素が必要です。 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