flusso tensoriale:: operazioni:: Mettere insieme

#include <array_ops.h>

Raccogli le fette dai params secondo indices .

Riepilogo

indices devono essere un tensore intero di qualsiasi dimensione (solitamente 0-D o ​​1-D). Produce un tensore di output con shape 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] , questa operazione permuterà params di conseguenza.

validate_indices : DEPRECATO. Se questa operazione è assegnata alla CPU, i valori negli indices vengono sempre convalidati in modo che rientrino nell'intervallo. Se assegnati alla GPU, gli indici fuori limite determinano un comportamento sicuro ma non specificato, che può 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 statiche pubbliche

ValidateIndices (bool x)

Strutture

tensorflow:: ops:: Gather:: Attrs

Setter attributi facoltativi per Gather .

Attributi pubblici

operazione

Operation operation

produzione

::tensorflow::Output output

Funzioni pubbliche

Mettere insieme

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

Mettere insieme

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

nodo

::tensorflow::Node * node() const 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const 

Funzioni statiche pubbliche

ConvalidaIndici

Attrs ValidateIndices(
  bool x
)