EnqueueTPUEmbeddingIntegerBatch

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
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.
statik EnqueueTPUEmbeddingIntegerBatch.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: "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.