Kumpulkan irisan dari sumbu `params` `sumbu` menurut `indeks`.
`indeks` harus berupa tensor bilangan bulat dengan dimensi apa pun (biasanya 0-D atau 1-D). Menghasilkan tensor keluaran dengan bentuk `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` di mana:
# Scalar indices (output is rank(params) - 1).
output[a_0, ..., a_n, b_0, ..., b_n] =
params[a_0, ..., a_n, indices, b_0, ..., b_n]
# Vector indices (output is rank(params)).
output[a_0, ..., a_n, i, b_0, ..., b_n] =
params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
# Higher rank indices (output is rank(params) + rank(indices) - 1).
output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
Perhatikan bahwa pada CPU, jika indeks di luar batas ditemukan, kesalahan akan dikembalikan. Pada GPU, jika indeks di luar batas ditemukan, angka 0 disimpan dalam nilai keluaran yang sesuai.
Lihat juga `tf.batch_gather` dan tf.gather_nd
.
Kelas Bersarang
kelas | Kumpulkan. Pilihan | Atribut opsional untuk Gather |
Metode Publik
Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
Gather.Options statis | batchDim (BatchDim panjang) |
statis <T, U memperluas Angka, V memperluas Angka> Kumpulkan <T> | |
Keluaran <T> | keluaran () Nilai dari `params` dikumpulkan dari indeks yang diberikan oleh `indices`, dengan bentuk `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`. |
Metode Warisan
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static Kumpulkan <T> buat ( Lingkup lingkup , parameter Operan <T>, indeks Operan <U>, sumbu Operan <V>, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi Gather baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
param | Tensor untuk mengumpulkan nilai. Setidaknya harus berada pada peringkat `sumbu + 1`. |
indeks | Tensor indeks. Harus berada dalam rentang `[0, params.shape[axis])`. |
sumbu | Sumbu di `params` untuk mengumpulkan `indeks`. Default ke dimensi pertama. Mendukung indeks negatif. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari Gather
Keluaran publik <T> keluaran ()
Nilai dari `params` dikumpulkan dari indeks yang diberikan oleh `indices`, dengan bentuk `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.