tensorflow :: ops :: DiagPart

#include <array_ops.h>

Restituisce la parte diagonale del tensore.

Sommario

Questa operazione restituisce un tensore con la parte diagonal input . La parte diagonal viene calcolata come segue:

Supponiamo che l' input abbia dimensioni [D1,..., Dk, D1,..., Dk] , l'output sia un tensore di rango k con dimensioni [D1,..., Dk] dove:

diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik] .

Per esempio:

# 'input' is [[1, 0, 0, 0]
              [0, 2, 0, 0]
              [0, 0, 3, 0]
              [0, 0, 0, 4]]

tf.diag_part(input) ==> [1, 2, 3, 4]

Argomenti:

  • scope: un oggetto Scope
  • input: Rango k tensore dove k è pari e diverso da zero.

Ritorna:

  • Output : la diagonale estratta.

Costruttori e distruttori

DiagPart (const :: tensorflow::Scope & scope, :: tensorflow::Input input)

Attributi pubblici

diagonal
operation

Funzioni pubbliche

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

Attributi pubblici

diagonale

::tensorflow::Output diagonal

operazione

Operation operation

Funzioni pubbliche

DiagPart

 DiagPart(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input
)

nodo

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const