このページは Cloud Translation API によって翻訳されました。

EnqueueTPUEmbeddingSparseBatch

パブリック最終クラスEnqueueTPUEmbeddingSparseBatch

SparseTensor からの TPUEmbedding 入力インデックスをキューに入れる操作。

この Op により、embedding_lookup_sparse() を使用するコードの移植が容易になりますが、トレーニングステップごとに許可される EnqueueTPUEmbeddingSparseBatch Op は 1 つだけであるため、この Op への引数を生成するには、embedding_lookup_sparse() への SparseTensor 引数の Python 前処理が必要になります。

3 つの入力リストの対応する位置にあるテンソルは同じ形状でなければなりません。つまり、対応する table_id によって記述されるテーブルへのルックアップの総数に等しい dim_size() を持つランク 1 でなければなりません。

ネストされたクラス

クラス EnqueueTPUEmbeddingSparseBatch.Options EnqueueTPUEmbeddingSparseBatchのオプション属性

定数

弦 OP_NAME TensorFlow コアエンジンによって認識される、この演算の名前

パブリックメソッド

静的EnqueueTPUEmbeddingSparseBatch.Options	コンバイナー(List<String> コンバイナー)
static <T extends TNumber 、U extends TNumber 、V extends TNumber > EnqueueTPUEmbeddingSparseBatch	create ( Scopeスコープ、Iterable< Operand <T>> sampleIndices、Iterable< Operand <U>> embeddingIndices、Iterable< Operand <V>> aggregationWeights、 Operand < TString > modeOverride、 Options... options) 新しい EnqueueTPUEmbeddingSparseBatch オペレーションをラップするクラスを作成するファクトリメソッド。
静的EnqueueTPUEmbeddingSparseBatch.Options	deviceOrdinal (長い deviceOrdinal)

継承されたメソッド

クラスorg.tensorflow.op.RawOpから

最終ブール値	等しい(オブジェクトオブジェクト)
最終整数	ハッシュコード（）
手術	オプ() この計算単位を単一の`Operation`として返します。
最後の文字列	toString ()

クラスjava.lang.Objectから

ブール値	等しい(オブジェクト arg0)
最終クラス<?>	getクラス()
整数	ハッシュコード（）
最後の空白	通知する()
最後の空白	すべて通知()
弦	toString ()
最後の空白	wait (long arg0, int arg1)
最後の空白	待機(長い引数0)
最後の空白	待って（）

インターフェースorg.tensorflow.op.Opから

抽象的な実行環境	環境() このオペレーションが作成された実行環境を返します。
抽象的な操作	オプ() この計算単位を単一の`Operation`として返します。

定数

パブリック静的最終文字列OP_NAME

TensorFlow コアエンジンによって認識される、この演算の名前

定数値: "EnqueueTPUEmbeddingSparseBatch"

パブリックメソッド

public static EnqueueTPUEmbeddingSparseBatch.Optionsコンバイナー(List<String> コンバイナー)

パラメーター

結合器	文字列スカラーのリスト。重み付き合計後の埋め込みアクティブ化を正規化する方法を指定する各埋め込みテーブルに 1 つあります。サポートされている結合子は、「mean」、「sum」、または「sqrtn」です。 'mean' の重みの合計が 0 になること、または 'sqrtn' の 2 乗重みの合計が 0 になることは無効です。コンバイナーが渡されない場合、デフォルトではすべてのテーブルに「sum」が使用されます。

public static EnqueueTPUEmbeddingSparseBatch create ( Scopeスコープ、Iterable< Operand <T>> sampleIndices、Iterable< Operand <U>> embeddingIndices、Iterable< Operand <V>> aggregationWeights、 Operand < TString > modeOverride、 Options...オプション)

新しい EnqueueTPUEmbeddingSparseBatch オペレーションをラップするクラスを作成するファクトリメソッド。

パラメーター

範囲	現在のスコープ
サンプルインデックス	対応するembedding_indicesとaggregation_weightsの値が属するトレーニング例と機能を指定するランク1テンソルのリスト。 Sample_indices[i] は b * nf + f に等しくなければなりません。ここで、nf は対応するテーブルの特徴の数、f は [0, nf) にあり、b は [0, バッチサイズ) にあります。
埋め込みインデックス	ランク 1 の Tensor、埋め込みテーブルへのインデックスのリスト。
集計重み	サンプルごと、つまり (トレーニング例、機能) ごとの集計重みを含むランク 1 テンソルのリスト。
モードオーバーライド	TPUEmbeddingConfiguration で指定されたモードをオーバーライドする文字列入力。サポートされている値は、{'unspecified'、'inference'、'training'、'backward_pass_only'} です。「unspecified」に設定すると、TPUEmbeddingConfiguration で設定されたモードが使用され、それ以外の場合は、mode_override が使用されます。
オプション	オプションの属性値を持ちます

戻り値

EnqueueTPUEmbeddingSparseBatch の新しいインスタンス

public static EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)

パラメーター

デバイス通常	使用する TPU デバイス。 0 以上で、ノードが配置されているタスク内の TPU コアの数未満である必要があります。