EnqueueTPUEmbeddingRaggedTensorBatch

genel final sınıfı EnqueueTPUEmbedddingRaggedTensorBatch

tf.nn.embedding_lookup() yöntemini kullanan kodun taşınmasını kolaylaştırır.

sample_splits[i], embedding_indices[i] ve aggregation_weights[i] i'inci özelliğe karşılık gelir. table_ids[i], bu özelliğe hangi yerleştirme tablosunun bakılacağını belirtir.

Giriş listelerinden ikisindeki (embedding_indices ve aggregation_weights) karşılık gelen konumlardaki tensörler aynı şekle sahip olmalıdır; yani ilgili özellik tarafından açıklanan tabloya yapılan toplam arama sayısına eşit dim_size() ile derece 1 olmalıdır.

İç İçe Sınıflar

sınıf EnqueueTPUEmbedddingRaggedTensorBatch.Options EnqueueTPUEmbeddingRaggedTensorBatch için isteğe bağlı özellikler

Genel Yöntemler

statik EnqueueTPUEmbedddingRaggedTensorBatch.Options
birleştiriciler (List<String> birleştiriciler)
static <T Sayıyı genişletir, U Sayıyı genişletir, V Sayıyı genişletir> EnqueueTPUEmbeddingRaggedTensorBatch
create ( Scopescope , Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Options... options)
Yeni bir EnqueueTPUEmbeddingRaggedTensorBatch işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
statik EnqueueTPUEmbedddingRaggedTensorBatch.Options
cihazSırası (Uzun cihazSırası)
statik EnqueueTPUEmbedddingRaggedTensorBatch.Options
maxSequenceLengths (Liste<Long> maxSequenceLengths)
statik EnqueueTPUEmbedddingRaggedTensorBatch.Options
numFeatures (Liste<Uzun> numFeatures)

Kalıtsal Yöntemler

Genel Yöntemler

public static EnqueueTPUEmbeddingRaggedTensorBatch.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 EnqueueTPUEmbeddingRaggedTensorBatch create ( Scopescope , Iterable< Operand <T>> sampleSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, List<Long> tableIds, Options.. . seçenekler)

Yeni bir EnqueueTPUEmbeddingRaggedTensorBatch işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.

Parametreler
kapsam mevcut kapsam
örnekBölmeler Gömme_indislerini ve toplama_ağırlıklarını satırlara bölmek için kırılma noktalarını belirten 1. Derece Tensörlerin listesi. ids bir RaggedTensor olduğunda, embedding_lookup() içindeki ids.row_splits'e karşılık gelir.
gömmeEndeksler Yerleştirme tablolarına eklenen 1. derece Tensörlerin listesi. ids bir RaggedTensor olduğunda, embedding_lookup() içindeki ids.values ​​değerine karşılık gelir.
toplamaAğırlıklar Eğitim başına örnek toplama ağırlıklarını içeren 1. Derece Tensörlerin listesi. Kimlikler bir RaggedTensor olduğunda, embedding_lookup() işlevindeki kimliklerle aynı row_splits'e sahip bir RaggedTensor'un değerler alanına karşılık gelir.
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.
tablo kimlikleri İlgili girişi aramak için yerleştirme tablosunun tanımlayıcısını belirten tam sayıların listesi (TPUEmbeddingConfiguration'daki TableDescriptor uzaklığı). i'inci giriş table_ids[i] kullanılarak aranır. table_ids listesinin boyutu sample_indices, embedding_indices ve aggregation_weights'ın boyutuna eşit olmalıdır.
seçenekler isteğe bağlı nitelik değerlerini taşır
İadeler
  • EnqueueTPUEmbeddingRaggedTensorBatch'in yeni bir örneği

genel statik EnqueueTPUEmbedddingRaggedTensorBatch.Options cihazıOrdinal (Uzun cihazOrdinal)

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.

public static EnqueueTPUEmbedddingRaggedTensorBatch.Options maxSequenceLengths (Liste<Long> maxSequenceLengths)

public static EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures (List<Long> numFeatures)