tensoreflusso:: ops:: Raccogliere

#include <array_ops.h>

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 è 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:

Ritorna:

  • Output : il tensore di uscita.

Costruttori e distruttori

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)

Attributi pubblici

operation
output

Funzioni pubbliche

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

Funzioni pubbliche statiche

ValidateIndices (bool x)

Strutture

tensorflow:: ops:: Gather:: Attrs

Setter di attributi facoltativi per Gather .

Attributi pubblici

operazione

Operation operation

produzione

::tensorflow::Output output

Funzioni pubbliche

Raccogliere

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

Raccogliere

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

nodo

::tensorflow::Node * node() const 

operatore::tensorflow::Input

 operator::tensorflow::Input() const 

operatore::tensorflow::Output

 operator::tensorflow::Output() const 

Funzioni pubbliche statiche

ValidateIndices

Attrs ValidateIndices(
  bool x
)