tensoreflusso:: ops:: MatrixSetDiag

#include <array_ops.h>

Restituisce un tensore di matrice in batch con nuovi valori diagonali in batch.

Riepilogo

Dati input e diagonal , questa operazione restituisce un tensore con la stessa forma e gli stessi valori di input , ad eccezione della diagonale principale delle matrici più interne. Questi verranno sovrascritti dai valori in diagonal .

L'output viene calcolato come segue:

Supponiamo che input abbia k+1 dimensioni [I, J, K, ..., M, N] e che diagonal abbia k dimensioni [I, J, K, ..., min(M, N)] . Allora l'output è un tensore di rango k+1 con dimensioni [I, J, K, ..., M, N] dove:

  • output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] per m == n .
  • output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] per m != n .

Argomenti:

  • scope: un oggetto Scope
  • input: Rango k+1 , dove k >= 1 .
  • diagonale: rango k , dove k >= 1 .

Ritorna:

  • Output : rango k+1 , con output.shape = input.shape .

Costruttori e distruttori

MatrixSetDiag (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input diagonal)

Attributi pubblici

operation
output

Funzioni pubbliche

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

Attributi pubblici

operazione

Operation operation

produzione

::tensorflow::Output output

Funzioni pubbliche

MatrixSetDiag

 MatrixSetDiag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input diagonal
)

nodo

::tensorflow::Node * node() const 

operatore::tensorflow::Input

 operator::tensorflow::Input() const 

operatore::tensorflow::Output

 operator::tensorflow::Output() const