TPU'da yerleştirme etkinleştirmelerini alan bir işlem.
TPU sistemi yerleştirme aramalarını ve toplamaları gerçekleştirir. Bu toplamaların sonuçları, bir XlaRecvTPUEmbeddingActivations Op'un çıktıları olarak Tensorflow Grafiğinde görülebilir. Bu op, modelde belirtilen tablo başına bir Tensör aktivasyon içeren bir liste döndürür.
Genel Yöntemler
statik XlaRecvTPUEkatıştırmaAktivasyonları | |
Yineleyici< İşlenen <Float>> | yineleyici () |
Liste< Çıkış <Float>> | çıktılar () Modeldeki gömme tablosu başına bir Tensör içeren gömme aktivasyonlarının bir TensorList'i. |
Kalıtsal Yöntemler
Genel Yöntemler
public static XlaRecvTPUEmbeddingActivations create ( Kapsam kapsamı, İşlenen <?> tekilleştirmeVerileri, Uzun numTable'lar, Dize yapılandırması)
Yeni bir XlaRecvTPUEmbeddingActivations işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|---|
tekilleştirmeVerileri | Tekilleştirme verilerini içeren =DT_VARIANT tipine sahip bir Tensör. Tensör, N öğe içeren bir XLA yuvalanmış demettir (burada N, TPU çipi başına gömme sayısının tensör çekirdeklerine oranıdır). İç içe geçmiş demetin her elemanı, 1. derece tensörlerden oluşan bir demettir. Her tensör ya TensorCore'a gömme araması için indeksler (DT_UINT32) ya da gömme arama işleminin çıktısına uygulanacak ağırlıklar (DT_FLOAT) içerir. |
numTable'lar | Çıkış aktivasyon tensörlerinin sayısı. Özellik tanımlayıcısı TPU yerleştirme yapılandırmasında mevcutsa, özelliklerin sayısına, aksi takdirde modeldeki yerleştirme tablolarının sayısına eşittir. |
yapılandırma | Serileştirilmiş TPUEmbedddingConfiguration protokolü. |
İadeler
- yeni bir XlaRecvTPUEmbeddingActivations örneği
genel Liste< Çıkış <Float>> çıkışlar ()
Modeldeki gömme tablosu başına bir Tensör içeren gömme aktivasyonlarının bir TensorList'i.