EnqueueTPUEmbeddingSparseBatch

kelas akhir publik EnqueueTPUEmbeddingSparseBatch

Sebuah operasi yang memasukkan indeks masukan TPUEmbedding dari SparseTensor.

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

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

Kelas Bersarang

kelas

EnqueueTPUEmbeddingSparseBatch.Options

Atribut opsional untuk EnqueueTPUEmbeddingSparseBatch

Metode Publik

EnqueueTPUEmbeddingSparseBatch.Options statis	penggabung (Daftar<String> penggabung)
statis <T memperluas Nomor, U memperluas Nomor, V memperluas Nomor> EnqueueTPUEmbeddingSparseBatch	buat ( Lingkup cakupan, 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 EnqueueTPUEmbeddingSparseBatch baru.
EnqueueTPUEmbeddingSparseBatch.Options statis	deviceOrdinal (Perangkat panjangOrdinal)

Metode Warisan

Dari kelas org.tensorflow.op.PrimitiveOp

boolean terakhir	sama dengan (Objek objek)
int terakhir	Kode hash ()
Operasi	op () Mengembalikan Operation yang mendasarinya
String terakhir	keString ()

Dari kelas java.lang.Object

boolean	sama dengan (Objek arg0)
Kelas terakhir<?>	dapatkan Kelas ()
ke dalam	Kode hash ()
kekosongan terakhir	memberitahu ()
kekosongan terakhir	beri tahuSemua ()
Rangkaian	keString ()
kekosongan terakhir	tunggu (arg0 panjang, int arg1)
kekosongan terakhir	tunggu (argumen panjang0)
kekosongan terakhir	Tunggu ()

Metode Publik

penggabung EnqueueTPUEmbeddingSparseBatch.Options statis publik (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 'kuadrat'. Jika penggabung tidak diteruskan, defaultnya adalah menggunakan 'jumlah' untuk semua tabel.

public static EnqueueTPUEmbeddingSparseBatch buat ( Lingkup cakupan , 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 EnqueueTPUEmbeddingSparseBatch baru.

Parameter

cakupan	ruang lingkup saat ini
sampelIndeks	Daftar Tensor peringkat 1 yang menentukan contoh dan fitur pelatihan yang memiliki nilai embedding_indices dan aggregation_weights yang sesuai. sample_indices[i] harus sama dengan b * nf + f, dengan nf adalah jumlah fitur dari tabel terkait, f di [0, nf), dan b di [0, ukuran batch).
menyematkan Indeks	Daftar Tensor peringkat 1, indeks ke dalam tabel penyematan.
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 TPUEmbeddingConfiguration. Nilai yang didukung adalah {'unspecified', 'inference', 'training', 'backward_pass_only'}. Jika disetel ke 'tidak ditentukan', mode yang disetel di TPUEmbeddingConfiguration akan digunakan, jika tidak, mode_override akan digunakan.
pilihan	membawa nilai atribut opsional

Kembali

• contoh baru EnqueueTPUEmbeddingSparseBatch

Enqueue statis publikTPUEmbeddingSparseBatch.Options deviceOrdinal (PerangkatOrdinal panjang)

Parameter

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