tensorflow :: opérations :: MatrixSetDiag

#include <array_ops.h>

Renvoie un tenseur de matrice par lots avec de nouvelles valeurs diagonales groupées.

Résumé

Compte tenu de l' input et de la diagonal , cette opération renvoie un tenseur avec la même forme et les mêmes valeurs que l' input , à l'exception de la diagonale principale des matrices les plus internes. Celles-ci seront écrasées par les valeurs en diagonal .

La sortie est calculée comme suit:

Supposons que l' input a k+1 dimensions [I, J, K, ..., M, N] et que la diagonal a k dimensions [I, J, K, ..., min(M, N)] . Alors la sortie est un tenseur de rang k+1 de dimensions [I, J, K, ..., M, N] où:

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

Arguments:

  • scope: un objet Scope
  • entrée: Rang k+1 , où k >= 1 .
  • diagonale: Rang k , où k >= 1 .

Retour:

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

Constructeurs et destructeurs

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

Attributs publics

operation
output

Fonctions publiques

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

Attributs publics

opération

Operation operation

production

::tensorflow::Output output

Fonctions publiques

MatrixSetDiag

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

nœud

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const