Memudahkan porting kode yang menggunakan tf.nn.embedding_lookup_sparse().
embedding_indices[i] dan aggregation_weights[i] sesuai dengan fitur ke-i.
Tensor pada posisi yang sesuai dalam tiga daftar masukan (sample_indices, embedding_indices, dan aggregation_weights) 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 | EnqueueTPUEmbeddingArbitraryTensorBatch.Options | Atribut opsional untuk EnqueueTPUEmbeddingArbitraryTensorBatch |
Metode Publik
EnqueueTPUEmbeddingArbitraryTensorBatch.Options statis | penggabung (Daftar<String> penggabung) |
statis <T memperluas Nomor, U memperluas Nomor, V memperluas Nomor> EnqueueTPUEmbeddingArbitraryTensorBatch | buat ( Cakupan cakupan , Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Opsi... opsi) Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingArbitraryTensorBatch baru. |
EnqueueTPUEmbeddingArbitraryTensorBatch.Options statis | deviceOrdinal (Perangkat panjangOrdinal) |
Metode Warisan
Metode Publik
penggabung EnqueueTPUEmbeddingArbitraryTensorBatch.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 EnqueueTPUEmbeddingArbitraryTensorBatch buat ( Lingkup cakupan , Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi EnqueueTPUEmbeddingArbitraryTensorBatch baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
sampleIndicesOrRowSplits | Daftar Tensor peringkat 2 yang menentukan contoh pelatihan yang memiliki nilai embedding_indices dan aggregation_weights yang sesuai. Jika ukuran dimensi pertamanya adalah 0, kami berasumsi setiap indeks_embes milik sampel yang berbeda. Baik int32 dan int64 diperbolehkan dan akan dikonversi ke int32 secara internal. Atau daftar Tensor peringkat 1 yang menentukan pembagian baris untuk memisahkan indeks_embedding dan bobot_agregasi menjadi beberapa baris. Ini sesuai dengan ids.row_splits di embedding_lookup(), ketika ids adalah RaggedTensor. Saat mengantrekan tensor ragged ND, hanya dimensi terakhir yang boleh di-ragged. pemisahan baris adalah tensor padat 1-D. Jika kosong, kami berasumsi tensor padat diteruskan ke op. Baik int32 dan int64 diizinkan dan akan dikonversi ke int32 secara internal. |
menyematkan Indeks | Daftar Tensor peringkat 1, indeks ke dalam tabel penyematan. Baik int32 dan int64 diperbolehkan dan akan dikonversi ke int32 secara internal. |
agregasiBerat | Daftar Tensor peringkat 1 yang berisi bobot agregasi per contoh pelatihan. Float32 dan float64 diperbolehkan dan akan dikonversi menjadi float32 secara internal. |
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 dari EnqueueTPUEmbeddingArbitraryTensorBatch
Enqueue statis publikTPUEmbeddingArbitraryTensorBatch.Options deviceOrdinal (Perangkat panjangOrdinal)
Parameter
perangkatOrdinal | Perangkat TPU yang akan digunakan. Harus >= 0 dan kurang dari jumlah inti TPU dalam tugas tempat node ditempatkan. |
---|