aliran tensor:: operasi:: Mengumpulkan

#include <array_ops.h>

Kumpulkan irisan dari params menurut indices .

Ringkasan

indices harus berupa tensor bilangan bulat dengan dimensi apa pun (biasanya 0-D atau 1-D). Menghasilkan tensor keluaran dengan indices.shape + params.shape[1:] di mana: 

    # Scalar indices
    output[:, ..., :] = params[indices, :, ... :]

    # Vector indices
    output[i, :, ..., :] = params[indices[i], :, ... :]

    # Higher rank indices
    output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]

Jika indices adalah permutasi dan len(indices) == params.shape[0] maka operasi ini akan mengubah params yang sesuai.

validate_indices : TIDAK DIGUNAKAN LAGI. Jika operasi ini ditugaskan ke CPU, nilai dalam indices selalu divalidasi agar berada dalam jangkauan. Jika ditetapkan ke GPU, indeks di luar batas menghasilkan perilaku yang aman namun tidak ditentukan, yang mungkin termasuk memunculkan kesalahan.

Argumen:

Pengembalian:

Konstruktor dan Destruktor

Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices)
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, const Gather::Attrs & attrs)

Atribut publik

operation
output

Fungsi publik

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Fungsi statis publik

ValidateIndices (bool x)

Struktur

tensorflow:: ops:: Kumpulkan:: Attrs

Penyetel atribut opsional untuk Gather .

Atribut publik

operasi 

Operation operation

keluaran 

::tensorflow::Output output

Fungsi publik

Mengumpulkan 

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices
)

Mengumpulkan 

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  const Gather::Attrs & attrs
)

simpul 

::tensorflow::Node * node() const

operator::tensorflow::Masukan 

 operator::tensorflow::Input() const

operator::tensorflow::Keluaran 

 operator::tensorflow::Output() const

Fungsi statis publik

Validasi Indeks

Attrs ValidateIndices(
  bool x
)