Renvoie la partie diagonale du tenseur.
Cette opération renvoie un tenseur avec la partie « diagonale » de « l'entrée ». La partie « diagonale » est calculée comme suit :
Supposons que `input` ait des dimensions `[D1,..., Dk, D1,..., Dk]`, alors la sortie est un tenseur de rang `k` avec des dimensions `[D1,..., Dk]` où:
`diagonale[i1,..., ik] = entrée[i1, ..., ik, i1,..., ik]`.
Par exemple :
# 'input' is [[1, 0, 0, 0]
[0, 2, 0, 0]
[0, 0, 3, 0]
[0, 0, 0, 4]]
tf.diag_part(input) ==> [1, 2, 3, 4]
Constantes
| Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
| Sortie <T> | comme Sortie () Renvoie le handle symbolique du tenseur. |
| statique <T étend TType > TensorDiagPart <T> | |
| Sortie <T> | diagonale () La diagonale extraite. |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
sortie publique <T> asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static TensorDiagPart <T> créer (portée d'étendue , entrée Operand <T>)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TensorDiagPart.
Paramètres
| portée | portée actuelle |
|---|---|
| saisir | Tenseur de rang k où k est pair et non nul. |
Retour
- une nouvelle instance de TensorDiagPart