genel son sınıf EnqueueTPUEmbeddingIntegerBatch
Giriş toplu tensörlerinin bir listesini TPUEmbedding'e sıralayan bir işlem.
İç İçe Sınıflar
sınıf | EnqueueTPUEmbedddingIntegerBatch.Options | EnqueueTPUEmbeddingIntegerBatch için isteğe bağlı özellikler |
Sabitler
Sicim | OP_NAME | Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı |
Genel Yöntemler
statik EnqueueTPUEmbedddingIntegerBatch | |
statik EnqueueTPUEmbeddingIntegerBatch.Options | cihazSırası (Uzun cihazSırası) |
Kalıtsal Yöntemler
boolean | eşittir (Nesne arg0) |
son Sınıf<?> | getClass () |
int | hash kodu () |
son boşluk | bildir () |
son boşluk | tümünü bildir () |
Sicim | toString () |
son boşluk | bekle (uzun arg0, int arg1) |
son boşluk | bekle (uzun arg0) |
son boşluk | Beklemek () |
özet Yürütme Ortamı | env () Bu operasyonun oluşturulduğu yürütme ortamını döndürün. |
soyut Operasyon |
Sabitler
genel statik son Dize OP_NAME
Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı
Sabit Değer: "EnqueueTPUEmbeddingIntegerBatch"
Genel Yöntemler
public static EnqueueTPUEmbeddingIntegerBatch create ( Kapsam kapsamı, Yinelenebilir< İşlenen < TInt32 >> toplu iş, İşlenen < TString > modeOverride, Seçenekler... seçenekler)
Yeni bir EnqueueTPUEmbeddingIntegerBatch işlemini saran bir sınıf oluşturmanın fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|---|
grup | Tablolardaki indeksleri içeren, her gömme tablosu için bir tane olmak üzere 1 boyutlu 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
- EnqueueTPUEmbeddingIntegerBatch'in yeni bir örneği
public static EnqueueTPUEmbeddingIntegerBatch.Options DeviceOrdinal (Uzun 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. |
---|