genel final sınıfı EnqueueTPUEyerleştirmeSparseBatch
Bir SparseTensor'dan TPUEmbedding giriş endekslerini sıraya koyan bir işlem.
Bu Op, embedding_lookup_sparse() kullanan kodun taşınmasını kolaylaştırır, ancak eğitim adımı başına yalnızca tek bir EnqueueTPUEmbeddingSparseBatch Op'a izin verildiğinden, bu Op'a argümanlar üretmek için SparseTensor argümanlarının embedding_lookup_sparse()'a yönelik bazı Python ön işlemesi gerekir.
Üç giriş listesinde karşılık gelen konumlardaki tensörler aynı şekle sahip olmalıdır; yani, ilgili table_id tarafından açıklanan tabloya yapılan toplam arama sayısına eşit dim_size() ile sıra 1 olmalıdır.
İç İçe Sınıflar
| sınıf | EnqueueTPUEmbedddingSparseBatch.Options | EnqueueTPUEmbeddingSparseBatch için isteğe bağlı özellikler | |
Sabitler
| Sicim | OP_NAME | Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı |
Genel Yöntemler
| statik EnqueueTPUEmbedddingSparseBatch.Options | birleştiriciler (List<String> birleştiriciler) |
| static <T TNumber'ı genişletir, U TNumber'ı genişletir, V TNumber'ı genişletir > EnqueueTPUEmbeddingSparseBatch | create ( Scopescope , Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, Options... options) Yeni bir EnqueueTPUEmbeddingSparseBatch işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi. |
| statik EnqueueTPUEmbedddingSparseBatch.Options | cihazSırası (Uzun cihazSırası) |
Kalıtsal Yöntemler
Sabitler
genel statik son Dize OP_NAME
Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı
Sabit Değer: "EnqueueTPUEmbeddingSparseBatch"
Genel Yöntemler
public static EnqueueTPUEmbeddingSparseBatch.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 değer tüm tablolar için 'toplam'ın kullanılmasıdır. |
|---|
public static EnqueueTPUEmbeddingSparseBatch create ( Kapsam kapsamı, Yinelenebilir< İşlenen <T>> sampleIndices, Yinelenebilir< İşlenen <U>> embeddingIndices, Yinelenebilir< İşlenen <V>> aggregationWeights, İşlenen < TString > modeOverride, Seçenekler... seçenekler)
Yeni bir EnqueueTPUEmbeddingSparseBatch işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
| kapsam | mevcut kapsam |
|---|---|
| örnekEndeksler | Karşılık gelen embedding_indices ve aggregation_weights değerlerinin ait olduğu eğitim örneğini ve özelliği belirten 1. Derece Tensörlerin listesi. sample_indices[i] b * nf + f'ye eşit olmalıdır; burada nf, karşılık gelen tablodaki özelliklerin sayısıdır, f, [0, nf)'dir ve b, [0, parti boyutu)'dur. |
| gömmeEndeksler | Yerleştirme tablolarına eklenen 1. derece Tensörlerin listesi. |
| toplamaAğırlıklar | Örnek başına (yani (eğitim örneği, özellik)) toplama ağırlıklarını içeren 1. Derece Tensörlerin listesi. |
| 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. |
| seçenekler | isteğe bağlı nitelik değerlerini taşır |
İadeler
- EnqueueTPUEmbeddingSparseBatch'in yeni bir örneği
public static EnqueueTPUEmbedddingSparseBatch.Options DeviceOrdinal (Long DeviceOrdinal)
Parametreler
| cihazSıra sayısı | Kullanılacak TPU cihazı. >= 0 olmalı ve düğümün yerleştirildiği görevdeki TPU çekirdek sayısından az olmalıdır. |
|---|