Gather

kelas akhir publik Berkumpul

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>
buat ( Lingkup lingkup, parameter Operan <T>, indeks Operan <U>, sumbu Operan <V>, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi Gather baru.
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 Gather.Options batchDims (BatchDims panjang)

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:]`.