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 |
Konstanta
Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolik tensor. |
Gather.Options statis | batchDim (BatchDim panjang) |
statis <T memperluas TType > 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
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
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 Gather <T> create ( Scope scope, Operand <T> params, Operand <? extends TNumber > indeks, Operand <? extends TNumber > axis, Options... options)
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:]`.