Merken Sie den Termin vor! Google I / O kehrt vom 18. bis 20. Mai zurück Registrieren Sie sich jetzt
Diese Seite wurde von der Cloud Translation API übersetzt.
Switch to English

Tensorflow :: ops :: MatrixSetDiag

#include <array_ops.h>

Gibt einen Batch-Matrix-Tensor mit neuen Batch-Diagonalwerten zurück.

Zusammenfassung

Bei gegebener input und diagonal gibt diese Operation einen Tensor mit derselben Form und denselben Werten wie die input , mit Ausnahme der Hauptdiagonale der innersten Matrizen. Diese werden durch die diagonal Werte überschrieben.

Die Ausgabe wird wie folgt berechnet:

Angenommen, die input hat k+1 Dimensionen [I, J, K, ..., M, N] und die diagonal hat k Dimensionen [I, J, K, ..., min(M, N)] . Dann ist die Ausgabe ein Tensor vom Rang k+1 mit den Dimensionen [I, J, K, ..., M, N] wobei:

  • output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] für m == n .
  • output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] für m != n .

Argumente:

  • scope: Ein Scope- Objekt
  • Eingabe: Rang k+1 , wobei k >= 1 .
  • Diagonale: Rang k , wobei k >= 1 .

Kehrt zurück:

  • Output : Rang k+1 , mit output.shape = input.shape .

Konstruktoren und Destruktoren

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

Öffentliche Attribute

operation
output

Öffentliche Funktionen

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

Öffentliche Attribute

Operation

Operation operation

Ausgabe

::tensorflow::Output output

Öffentliche Funktionen

MatrixSetDiag

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

Knoten

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const