aliran tensor:: operasi:: Mengumpulkan

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:

• ruang lingkup: Objek Lingkup

Pengembalian:

• Output : Tensor keluaran.

Atribut publik

Operation operation

::tensorflow::Output output

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs ValidateIndices(
  bool x
)