EnqueueTPUEmbeddingBatch

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.