tf.nn.embedding_lookup_sparse() yöntemini kullanan kodun taşınmasını kolaylaştırır.
embedding_indices[i] ve aggregation_weights[i] i'inci özelliğe karşılık gelir.
Üç 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 | DynamicEnqueueTPUEmbedddingArbitraryTensorBatch.Options | DynamicEnqueueTPUEmbeddingArbitraryTensorBatch için isteğe bağlı özellikler |
Genel Yöntemler
statik DynamicEnqueueTPUEmbedddingArbitraryTensorBatch.Options | birleştiriciler (List<String> birleştiriciler) |
static <T Sayıyı genişletir, U Sayıyı genişletir, V Sayıyı genişletir> DynamicEnqueueTPUEmbeddingArbitraryTensorBatch | create ( Scopescope , Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Operand <Integer> DeviceOrdinal, Options... options) Yeni bir DynamicEnqueueTPUEmbeddingArbitraryTensorBatch işlemini saran bir sınıf oluşturmanın fabrika yöntemi. |
Kalıtsal Yöntemler
Genel Yöntemler
public static DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.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 DynamicEnqueueTPUEmbeddingArbitraryTensorBatch create ( Scopescope , Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Operand <Integer> DeviceOrdinal, Options.. . seçenekler)
Yeni bir DynamicEnqueueTPUEmbeddingArbitraryTensorBatch işlemini saran bir sınıf oluşturmanın fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|---|
sampleIndicesOrRowSplits | Karşılık gelen embedding_indices ve aggregation_weights değerlerinin ait olduğu eğitim örneğini belirten 2. Derece Tensörlerin listesi. İlk boyutunun boyutu 0 ise her embedding_index'in farklı bir örneğe ait olduğunu varsayarız. Hem int32 hem de int64'e izin verilir ve dahili olarak int32'ye dönüştürülür. Veya embedding_indices ve aggregation_weights'ı satırlara bölmek için satır bölümlerini belirten 1. Derece Tensörlerin bir listesi. ids bir RaggedTensor olduğunda, embedding_lookup() içindeki ids.row_splits'e karşılık gelir. ND düzensiz tensörünü kuyruğa alırken yalnızca son boyutun düzensiz olmasına izin verilir. satır bölmeleri 1 boyutlu yoğun tensördür. Boş olduğunda yoğun bir tensörün op'a iletildiğini varsayarız Hem int32'ye hem de int64'e izin verilir ve dahili olarak int32'ye dönüştürülür. |
gömmeEndeksler | Yerleştirme tablolarına eklenen 1. derece Tensörlerin listesi. Hem int32 hem de int64'e izin verilir ve dahili olarak int32'ye dönüştürülür. |
toplamaAğırlıklar | Eğitim başına örnek toplama ağırlıklarını içeren 1. Derece Tensörlerin listesi. Hem float32 hem de float64'e izin verilir ve dahili olarak float32'ye dönüştürülecektir. |
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. |
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. |
seçenekler | isteğe bağlı nitelik değerlerini taşır |
İadeler
- DynamicEnqueueTPUEmbeddingArbitraryTensorBatch'in yeni bir örneği