tensoreflusso:: ops:: Raccogliere

Raccogli le sezioni dai params in base agli indices .

Riepilogo

indices devono essere un tensore intero di qualsiasi dimensione (solitamente 0-D o ​​1-D). Produce un tensore di output con forma indices.shape + params.shape[1:] dove:

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

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

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

Se indices è una permutazione e len(indices) == params.shape[0] allora questa operazione permuterà params di conseguenza.

validate_indices : DEPRECATO. Se questa operazione viene assegnata alla CPU, i valori negli indices vengono sempre convalidati per rientrare nell'intervallo. Se assegnati alla GPU, gli indici fuori limite determinano un comportamento sicuro ma non specificato, che potrebbe includere la generazione di un errore.

Argomenti:

• scope: un oggetto Scope

Resi:

• Output : il tensore di uscita.

Attributi pubblici

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
)