tensoreflusso:: ops:: MatrixSetDiag

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 .

Resi:

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

Attributi pubblici

Operation operation

::tensorflow::Output output

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const