Tensorflow :: ops :: Versammeln

#include <array_ops.h>

Sammeln Sie Scheiben von params gemäß den indices .

Zusammenfassung

indices müssen ein ganzzahliger Tensor einer beliebigen Dimension sein (normalerweise 0-D oder 1-D). Erzeugt einen Ausgangstensor mit indices.shape + params.shape[1:] wobei:

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

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

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

Wenn indices eine Permutation und len(indices) == params.shape[0] sind, permutiert diese Operation die params entsprechend.

validate_indices : DEPRECATED. Wenn diese Operation der CPU zugewiesen ist, werden Werte in indices immer so validiert, dass sie innerhalb des Bereichs liegen. Bei Zuweisung zur GPU führen nicht gebundene Indizes zu einem sicheren, aber nicht spezifizierten Verhalten, zu dem auch das Auslösen eines Fehlers gehören kann.

Argumente:

Kehrt zurück:

  • Output : Der Ausgangstensor.

Konstruktoren und Destruktoren

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)

Öffentliche Attribute

operation
output

Öffentliche Funktionen

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

Öffentliche statische Funktionen

ValidateIndices (bool x)

Strukturen

tensorflow :: ops :: Gather :: Attrs

Optionale Attributsetzer fürGather .

Öffentliche Attribute

Operation

Operation operation

Ausgabe

::tensorflow::Output output

Öffentliche Funktionen

Versammeln

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

Versammeln

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

Knoten

::tensorflow::Node * node() const

operator :: tensorflow :: Input

 operator::tensorflow::Input() const

operator :: tensorflow :: Output

 operator::tensorflow::Output() const

Öffentliche statische Funktionen

ValidateIndices

Attrs ValidateIndices(
  bool x
)