public class akhir EnqueueTPUEmbeddingSparseTensorBatch
Memudahkan porting kode yang menggunakan tf.nn.embedding_lookup_sparse().
sample_indices[i], embedding_indices[i] dan aggregation_weights[i] sesuai dengan fitur ke-i. table_ids[i] menunjukkan tabel penyematan mana yang mencari fitur ke-i.
Tensor pada posisi terkait dalam tiga daftar input (indeks_sampel, indeks_penyematan, dan bobot_agregasi) harus memiliki bentuk yang sama, yaitu peringkat 1 dengan dim_size() sama dengan jumlah total pencarian ke dalam tabel yang dijelaskan oleh fitur terkait.
Kelas Bersarang
| kelas | EnqueueTPUEbeddingSparseTensorBatch.Options | Atribut opsional untuk EnqueueTPUEmbeddingSparseTensorBatch | |
Metode Publik
| statis EnqueueTPUEmbeddingSparseTensorBatch.Options | combiners (Daftar <String> combiners) |
| statis <T meluas Nomor, U meluas Nomor, V meluas Nomor> EnqueueTPUEmbeddingSparseTensorBatch | membuat ( Lingkup lingkup, Iterable < Operan <T >> sampleIndices, Iterable < Operan <U >> embeddingIndices, Iterable < Operan <V >> aggregationWeights, Operan <String> modeOverride, Daftar <Panjang> tableIds, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingSparseTensorBatch baru. |
| statis EnqueueTPUEmbeddingSparseTensorBatch.Options | deviceOrdinal (Long deviceOrdinal) |
| statis EnqueueTPUEmbeddingSparseTensorBatch.Options | maxSequenceLengths (Daftar <Panjang> maxSequenceLengths) |
| statis EnqueueTPUEmbeddingSparseTensorBatch.Options | numFeatures (Daftar <Panjang> numFeatures) |
Metode yang Diwarisi
Metode Publik
public static EnqueueTPUEmbeddingSparseTensorBatch.Options combiners (Daftar <String> combiners)
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 EnqueueTPUEmbeddingSparseTensorBatch membuat ( Scope lingkup, Iterable < Operan <T >> sampleIndices, Iterable < Operan <U >> embeddingIndices, Iterable < Operan <V >> aggregationWeights, Operan <String> modeOverride, Daftar <Panjang> tableIds, Pilihan .. . Pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingSparseTensorBatch baru.
Parameter
| cakupan | lingkup saat ini |
|---|---|
| sampelIndeks | Daftar Tensor peringkat 1 yang menentukan contoh pelatihan yang memiliki nilai embedding_indices dan aggregation_weights yang sesuai. Ini sesuai dengan sp_ids.indices[:,0] di embedding_lookup_sparse(). |
| embeddingIndex | Daftar Tensor peringkat 1, indeks ke dalam tabel embedding. Ini sesuai dengan sp_ids.values di embedding_lookup_sparse(). |
| agregasiBerat | Daftar Tensor peringkat 1 yang berisi bobot agregasi contoh per pelatihan. Ini sesuai dengan sp_weights.values di embedding_lookup_sparse(). |
| modeOverride | Input string yang menggantikan mode yang ditentukan dalam TPUEmbeddingConfiguration. Nilai yang didukung adalah {'unspecified', 'inference', 'training', 'backward_pass_only'}. Ketika disetel ke 'tidak ditentukan', mode yang diatur dalam TPUEmbeddingConfiguration digunakan, jika tidak, mode_override digunakan. |
| tableIds | Daftar bilangan bulat yang menentukan pengidentifikasi tabel penyematan (offset dari TableDescriptor di TPUembeddingConfiguration) untuk mencari input yang sesuai. Input ke-i dicari menggunakan table_ids[i]. Ukuran daftar table_ids harus sama dengan ukuran sample_indices, embedding_indices, dan aggregation_weights. |
| pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari EnqueueTPUEmbeddingSparseTensorBatch
public static EnqueueTPUEmbeddingSparseTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
Parameter
| perangkatOrdinal | Perangkat TPU yang akan digunakan. Harus >= 0 dan kurang dari jumlah inti TPU dalam tugas di mana node ditempatkan. |
|---|