tensorflow :: opérations :: MatrixSetDiag

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 .

Attributs publics

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