Kumpulkan irisan dari sumbu `params` `sumbu` menurut `indeks`.
`indeks` harus berupa tensor bilangan bulat dari 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, 0 disimpan dalam nilai output 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 simbolis tensor. |
Pengumpulan statis.Opsi | batchDims (BatchDims panjang) |
statis <T, U memperpanjang Angka, V memperpanjang 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 yang Diwarisi
Metode Publik
Output publik <T> sebagai Output ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static Kumpulkan <T> buat ( Lingkup lingkup, parameter Operand <T>, indeks Operand <U>, sumbu Operand <V>, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi Gather baru.
Parameter
cakupan | lingkup saat ini |
---|---|
params | Tensor untuk mengumpulkan nilai. Harus setidaknya peringkat `sumbu + 1`. |
indeks | Tensor indeks. Harus 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:]`.
Kumpulkan irisan dari sumbu `params` `sumbu` menurut `indeks`.
`indeks` harus berupa tensor bilangan bulat dari 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, 0 disimpan dalam nilai output 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 simbolis tensor. |
Pengumpulan statis.Opsi | batchDims (BatchDims panjang) |
statis <T, U memperpanjang Angka, V memperpanjang 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 yang Diwarisi
Metode Publik
Output publik <T> sebagai Output ()
Mengembalikan pegangan simbolis tensor.
Input ke operasi TensorFlow adalah output dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static Kumpulkan <T> buat ( Lingkup lingkup, parameter Operand <T>, indeks Operand <U>, sumbu Operand <V>, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi Gather baru.
Parameter
cakupan | lingkup saat ini |
---|---|
params | Tensor untuk mengumpulkan nilai. Harus setidaknya peringkat `sumbu + 1`. |
indeks | Tensor indeks. Harus 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:]`.