genel final sınıfı EnqueueTPUEmbeddingBatch
Giriş toplu tensörlerinin bir listesini TPUEmbedding'e sıralayan bir işlem.
Giriş toplu tensörlerinin bir listesini TPUEmbedding'e sıralayan bir işlem.
İç İçe Sınıflar
sınıf | EnqueueTPUEmbedddingBatch.Options | EnqueueTPUEmbeddingBatch için isteğe bağlı özellikler |
Genel Yöntemler
statik EnqueueTPUEmbeddingBatch.Options | birleştiriciler (List<String> birleştiriciler) |
statik EnqueueTPUEkatıştırmaToplusu | create ( Kapsam kapsamı, Yinelenebilir< İşlenen <Dize>> toplu iş, İşlenen <Dize> modeOverride, Seçenekler... seçenekler) Yeni bir EnqueueTPUEmbeddingBatch işlemini saran bir sınıf oluşturmanın fabrika yöntemi. |
statik EnqueueTPUEmbeddingBatch.Options | cihazSırası (Uzun cihazSırası) |
Kalıtsal Yöntemler
Genel Yöntemler
public static EnqueueTPUEmbeddingBatch.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 EnqueueTPUEmbeddingBatch create ( Kapsam kapsamı, Yinelenebilir< İşlenen <Dize>> toplu iş, İşlenen <Dize> modeOverride, Seçenekler... seçenekler)
Yeni bir EnqueueTPUEmbeddingBatch işlemini saran bir sınıf oluşturmanın fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|---|
grup | Dist_belief.SparseFeatures protokolleri olarak kodlanmış toplu girişleri içeren, her gömme tablosu için bir tane olmak üzere 1 boyutlu tensörlerin listesi. SparseFeatures protokolündeki ağırlık alanı bir kimlik için doldurulmamışsa ağırlığın 1,0 olduğu varsayılır. |
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
- EnqueueTPUEmbeddingBatch'in yeni bir örneği
public static EnqueueTPUEmbeddingBatch.Options DeviceOrdinal (Uzun DeviceOrdinal)
Parametreler
cihazSıra sayısı | Kullanılacak TPU cihazı. Op bir TPU cihazında çalışırken bu -1 olmalıdır ve Op, CPU cihazında çalışırken >= 0 olmalıdır. |
---|