XlaRecvTPUEmbeddingActivations

genel son sınıf XlaRecvTPUEmbeddingActivations

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ı
create ( Kapsam kapsamı, İşlenen <?> veri tekilleştirmeVerileri, Uzun numTable'lar, String yapılandırması)
Yeni bir XlaRecvTPUEmbeddingActivations işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Yineleyici< İşlenen <Float>>
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

public Yineleyici< İşlenen <Float>> yineleyici ()

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.