パブリック最終クラスEnqueueTPUEmbeddingRaggedTensorBatch
tf.nn.embedding_lookup() を使用するコードの移植を容易にします。
Sample_splits[i]、embedding_indices[i]、および aggregation_weights[i] が i 番目の特徴に対応します。 table_ids[i] は、i 番目の特徴を検索するための埋め込みテーブルを示します。
入力リストのうち 2 つ (embedding_indices と aggregation_weights) の対応する位置にあるテンソルは、同じ形状、つまり対応する特徴によって記述されるテーブルへのルックアップの総数に等しい dim_size() を持つランク 1 を持つ必要があります。
ネストされたクラス
クラス | EnqueueTPUEmbeddingRaggedTensorBatch.Options | EnqueueTPUEmbeddingRaggedTensorBatch のオプションの属性 |
パブリックメソッド
静的EnqueueTPUEmbeddingRaggedTensorBatch.Options | コンバイナー(List<String> コンバイナー) |
static <T extends Number, U extends Number, V extends Number> EnqueueTPUEmbeddingRaggedTensorBatch | |
静的EnqueueTPUEmbeddingRaggedTensorBatch.Options | deviceOrdinal (長い deviceOrdinal) |
静的EnqueueTPUEmbeddingRaggedTensorBatch.Options | maxSequenceLengths (List<Long> maxSequenceLengths) |
静的EnqueueTPUEmbeddingRaggedTensorBatch.Options | numFunctions (List<Long> numFunctions) |
継承されたメソッド
パブリックメソッド
public static EnqueueTPUEmbeddingRaggedTensorBatch.Optionsコンバイナー(List<String> コンバイナー)
パラメーター
結合器 | 文字列スカラーのリスト。重み付き合計後の埋め込みアクティブ化を正規化する方法を指定する各埋め込みテーブルに 1 つあります。サポートされている結合子は、「mean」、「sum」、または「sqrtn」です。 'mean' の重みの合計が 0 になること、または 'sqrtn' の 2 乗重みの合計が 0 になることは無効です。コンバイナーが渡されない場合、デフォルトではすべてのテーブルに「sum」が使用されます。 |
---|
public static EnqueueTPUEmbeddingRaggedTensorBatch create ( Scopeスコープ、Iterable< Operand <T>> sampleSplits、Iterable< Operand <U>> embeddingIndices、Iterable< Operand <V>> aggregationWeights、 Operand <String> modeOverride、List<Long> tableIds、 Options.. .オプション)
新しい EnqueueTPUEmbeddingRaggedTensorBatch オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
サンプル分割 | embedding_indices と aggregation_weights を行に分割するためのブレーク ポイントを指定するランク 1 Tensor のリスト。 ids が RaggedTensor の場合、embedding_lookup() の ids.row_splits に対応します。 |
埋め込みインデックス | ランク 1 の Tensor、埋め込みテーブルへのインデックスのリスト。 ids が RaggedTensor の場合、embedding_lookup() の ids.values に対応します。 |
集計重み | トレーニング サンプルごとの集約重みを含むランク 1 テンソルのリスト。 ids が RaggedTensor の場合、embedding_lookup() の ids と同じ row_splits を持つ RaggedTensor の値フィールドに対応します。 |
モードオーバーライド | TPUEmbeddingConfiguration で指定されたモードをオーバーライドする文字列入力。サポートされている値は、{'unspecified'、'inference'、'training'、'backward_pass_only'} です。 「unspecified」に設定すると、TPUEmbeddingConfiguration で設定されたモードが使用され、それ以外の場合は、mode_override が使用されます。 |
テーブルID | 対応する入力を検索するための埋め込みテーブルの識別子 (TPUEmbeddingConfiguration 内の TableDescriptor のオフセット) を指定する整数のリスト。 i 番目の入力は、table_ids[i] を使用して検索されます。 table_ids リストのサイズは、sample_indices、embedding_indices、および aggregation_weights のサイズと等しくなければなりません。 |
オプション | オプションの属性値を持ちます |
戻り値
- EnqueueTPUEmbeddingRaggedTensorBatch の新しいインスタンス
public static EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
パラメーター
デバイス通常 | 使用する TPU デバイス。 0 以上で、ノードが配置されているタスク内の TPU コアの数未満である必要があります。 |
---|