genel final sınıfı EnqueueTPUEmbedddingRaggedTensorBatch
tf.nn.embedding_lookup() yöntemini kullanan kodun taşınmasını kolaylaştırır.
sample_splits[i], embedding_indices[i] ve aggregation_weights[i] i'inci özelliğe karşılık gelir. table_ids[i], bu özelliğe hangi yerleştirme tablosunun bakılacağını belirtir.
Giriş listelerinden ikisindeki (embedding_indices ve aggregation_weights) karşılık gelen konumlardaki tensörler aynı şekle sahip olmalıdır; yani ilgili özellik tarafından açıklanan tabloya yapılan toplam arama sayısına eşit dim_size() ile derece 1 olmalıdır.
İç İçe Sınıflar
| sınıf | EnqueueTPUEmbedddingRaggedTensorBatch.Options | EnqueueTPUEmbeddingRaggedTensorBatch için isteğe bağlı özellikler | |
Sabitler
| Sicim | OP_NAME | Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı |
Genel Yöntemler
| statik EnqueueTPUEmbedddingRaggedTensorBatch.Options | birleştiriciler (List<String> birleştiriciler) |
| static <T, TNumber'ı genişletir, U, TNumber'ı genişletir, V, TNumber'ı genişletir > EnqueueTPUEmbeddingRaggedTensorBatch | create ( Scopescope , Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options... options) Yeni bir EnqueueTPUEmbeddingRaggedTensorBatch işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi. |
| statik EnqueueTPUEmbedddingRaggedTensorBatch.Options | cihazSırası (Uzun cihazSırası) |
| statik EnqueueTPUEmbedddingRaggedTensorBatch.Options | maxSequenceLengths (Liste<Long> maxSequenceLengths) |
Kalıtsal Yöntemler
Sabitler
genel statik son Dize OP_NAME
Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı
Sabit Değer: "EnqueueTPUEmbeddingRaggedTensorBatch"
Genel Yöntemler
public static EnqueueTPUEmbeddingRaggedTensorBatch.Options birleştiricileri (List<String> birleştiricileri)
Parametreler
| birleştiriciler | Ağırlıklı toplamdan sonra gömme aktivasyonlarının nasıl normalleştirileceğini belirten, her gömme tablosu için bir tane olmak üzere dize skalerlerinin bir listesi. Desteklenen birleştiriciler 'ortalama', 'toplam' veya 'sqrtn'dir. 'Mean' için ağırlıkların toplamının 0 olması veya 'sqrtn' için ağırlıkların karelerinin toplamının 0 olması geçersizdir. Birleştiriciler aktarılmazsa, varsayılan olarak tüm tablolar için 'toplam' kullanılması gerekir. |
|---|
public static EnqueueTPUEmbeddingRaggedTensorBatch create ( Scopescope , Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options.. . seçenekler)
Yeni bir EnqueueTPUEmbeddingRaggedTensorBatch işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
| kapsam | mevcut kapsam |
|---|---|
| örnekBölmeler | Gömme_indislerini ve toplama_ağırlıklarını satırlara bölmek için kırılma noktalarını belirten 1. Derece Tensörlerin listesi. ids bir RaggedTensor olduğunda, embedding_lookup() içindeki ids.row_splits'e karşılık gelir. |
| gömmeEndeksler | Yerleştirme tablolarına eklenen 1. derece Tensörlerin listesi. ids bir RaggedTensor olduğunda, embedding_lookup() içindeki ids.values değerine karşılık gelir. |
| toplamaAğırlıklar | Eğitim başına örnek toplama ağırlıklarını içeren 1. Derece Tensörlerin listesi. Kimlikler bir RaggedTensor olduğunda, embedding_lookup() işlevindeki kimliklerle aynı row_splits'e sahip bir RaggedTensor'un değerler alanına karşılık gelir. |
| moduGeçersiz kıl | TPUEmbeddingConfiguration'da belirtilen modu geçersiz kılan bir dize girişi. Desteklenen değerler şunlardır: {'belirtilmemiş', 'çıkarım', 'eğitim', 'backward_pass_only'}. 'Belirtilmemiş' olarak ayarlandığında, TPUEmbeddingConfiguration'da ayarlanan mod kullanılır, aksi takdirde mode_override kullanılır. |
| tablo kimlikleri | İlgili girişi aramak için yerleştirme tablosunun tanımlayıcısını belirten tam sayıların listesi (TPUEmbeddingConfiguration'daki TableDescriptor uzaklığı). i'inci giriş table_ids[i] kullanılarak aranır. table_ids listesinin boyutu sample_indices, embedding_indices ve aggregation_weights'ın boyutuna eşit olmalıdır. |
| seçenekler | isteğe bağlı nitelik değerlerini taşır |
İadeler
- EnqueueTPUEmbeddingRaggedTensorBatch'in yeni bir örneği
genel statik EnqueueTPUEmbedddingRaggedTensorBatch.Options cihazıOrdinal (Uzun cihazOrdinal)
Parametreler
| cihazSırası | Kullanılacak TPU cihazı. >= 0 olmalı ve düğümün yerleştirildiği görevdeki TPU çekirdek sayısından az olmalıdır. |
|---|