EnqueueTPUEmbeddingSparseTensorBatch

genel son sınıf EnqueueTPUEyerleştirmeSparseTensorBatch

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

sample_indices[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.

Üç girdi listesindeki (örnek_indices, 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 EnqueueTPUEmbedddingSparseTensorBatch.Options EnqueueTPUEmbeddingSparseTensorBatch için isteğe bağlı özellikler

Sabitler

Sicim OP_NAME Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı

Genel Yöntemler

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

Kalıtsal Yöntemler

Sabitler

genel statik son Dize OP_NAME

Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı

Sabit Değer: "EnqueueTPUEmbeddingSparseTensorBatch"

Genel Yöntemler

public static EnqueueTPUEmbeddingSparseTensorBatch.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 olarak tüm tablolar için 'toplam' kullanılması gerekir.

public static EnqueueTPUEmbeddingSparseTensorBatch create ( Scopescope , Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, List<Long> tableIds, Options.. . seçenekler)

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

Parametreler
kapsam mevcut kapsam
örnekEndeksler Karşılık gelen embedding_indices ve aggregation_weights değerlerinin ait olduğu eğitim örneğini belirten 1. Derece Tensörlerin listesi. embedding_lookup_sparse() içindeki sp_ids.indices[:,0] öğesine karşılık gelir.
gömmeEndeksler Yerleştirme tablolarına eklenen 1. derece Tensörlerin listesi. embedding_lookup_sparse() içindeki sp_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. embedding_lookup_sparse() içindeki sp_weights.values ​​değerine 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
  • EnqueueTPUEmbeddingSparseTensorBatch'in yeni bir örneği

public static EnqueueTPUEmbedddingSparseTensorBatch.Options cihazıOrdinal (Uzun cihazOrdinal)

Parametreler
cihazSırası 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 EnqueueTPUEmbedddingSparseTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)