EnqueueTPUEmbeddingSparseBatch

kelas akhir publik EnqueueTPUEmbeddingSparseBatch

Sebuah op yang mengantrekan indeks input TPUEbedding dari SparseTensor.

Op ini memudahkan porting kode yang menggunakan embedding_lookup_sparse(), meskipun beberapa prapemrosesan Python dari argumen SparseTensor ke embedding_lookup_sparse() diperlukan untuk menghasilkan argumen ke Op ini, karena hanya satu EnqueueTPUEmbeddingSparseBatch Op yang diizinkan per langkah pelatihan.

Tensor pada posisi yang sesuai dalam tiga daftar input harus memiliki bentuk yang sama, yaitu peringkat 1 dengan dim_size() sama dengan jumlah total pencarian ke dalam tabel yang dijelaskan oleh table_id yang sesuai.

Kelas Bersarang

kelas

EnqueueTPUEbeddingSparseBatch.Options

Atribut opsional untuk EnqueueTPUEmbeddingSparseBatch

Metode Publik

EnqueueTPUEmbeddingSparseBatch.Options . statis	penggabung (penggabung Daftar<String>)
static <T memperpanjang Angka, U memperpanjang Angka, V memperpanjang Angka> EnqueueTPUEmbeddingSparseBatch	create ( Scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> agregasiWeights, Operand <String> modeOverride, Options... options) Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbbeddingSparseBatch baru.
EnqueueTPUEmbeddingSparseBatch.Options . statis	deviceOrdinal (Long deviceOrdinal)

Metode yang Diwarisi

Dari kelas org.tensorflow.op.PrimitiveOp

boolean akhir	sama (objek objek)
akhir	kode hash ()
Operasi	op () Mengembalikan Operation yang mendasarinya
string terakhir	toString ()

Dari kelas java.lang.Object

boolean	sama dengan (Objek arg0)
Kelas terakhir<?>	dapatkanKelas ()
ke dalam	kode hash ()
kekosongan akhir	beri tahu ()
kekosongan akhir	beri tahu Semua ()
Rangkaian	toString ()
kekosongan akhir	tunggu (arg panjang0, int arg1)
kekosongan akhir	tunggu (arg panjang0)
kekosongan akhir	tunggu ()

Metode Publik

Penggabung EnqueueTPUEmbeddingSparseBatch.Options publik statis ( penggabung Daftar<String>)

Parameter

penggabung	Daftar skalar string, satu untuk setiap tabel penyematan yang menentukan cara menormalkan aktivasi penyematan setelah penjumlahan berbobot. Penggabung yang didukung adalah 'mean', 'sum', atau 'sqrtn'. Tidak valid jika jumlah bobot menjadi 0 untuk 'rata-rata' atau jumlah bobot kuadrat menjadi 0 untuk 'persegi'. Jika penggabung tidak lulus, defaultnya adalah menggunakan 'jumlah' untuk semua tabel.

public static EnqueueTPUEmbeddingSparseBatch create ( Scope scope, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbbeddingSparseBatch baru.

Parameter

cakupan	lingkup saat ini
sampelIndeks	Daftar Tensor peringkat 1 yang menentukan contoh pelatihan dan fitur yang memiliki nilai embedding_indices dan aggregation_weights yang sesuai. sample_indices[i] harus sama dengan b * nf + f, di mana nf adalah jumlah fitur dari tabel yang sesuai, f dalam [0, nf), dan b dalam [0, ukuran batch).
embeddingIndex	Daftar Tensor peringkat 1, indeks ke dalam tabel embedding.
agregasiBerat	Daftar Tensor peringkat 1 yang berisi per sampel -- yaitu per (contoh pelatihan, fitur) -- bobot agregasi.
modeOverride	Input string yang menggantikan mode yang ditentukan dalam TPUEmbbeddingConfiguration. Nilai yang didukung adalah {'unspecified', 'inference', 'training', 'backward_pass_only'}. Ketika diatur ke 'tidak ditentukan', mode yang diatur dalam TPUEmbeddingConfiguration digunakan, jika tidak, mode_override digunakan.
pilihan	membawa nilai atribut opsional

Kembali

• contoh baru dari EnqueueTPUEmbeddingSparseBatch

perangkat EnqueueTPUEmbbeddingSparseBatch.Options publik statis Ordinal (perangkat panjang Ordinal)

Parameter

perangkatOrdinal	Perangkat TPU yang akan digunakan. Harus >= 0 dan kurang dari jumlah inti TPU dalam tugas di mana node ditempatkan.