MatMul

cours final public MatMul

Multipliez la matrice "a" par la matrice "b".

Les entrées doivent être des matrices bidimensionnelles et la dimension intérieure de "a" (après avoir été transposée si transpose_a est vrai) doit correspondre à la dimension extérieure de "b" (après avoir été transposée si transposed_b est vrai).

Remarque : L'implémentation du noyau par défaut pour MatMul sur les GPU utilise cublas.

Classes imbriquées

classe MatMul.Options Attributs facultatifs pour MatMul

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 > MatMul <T>
créer ( Portée de portée , Opérande <T> a, Opérande <T> b, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération MatMul.
Sortie <T>
MatMul.Options statique
transposeA (transposeA booléenne)
MatMul.Options statique
transposeB (transposeB booléenne)

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

Valeur constante : "MatMul"

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 MatMul <T> create ( Portée de la portée , Opérande <T> a, Opérande <T> b, Options... options)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération MatMul.

Paramètres
portée portée actuelle
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de MatMul

public Sortie <T> produit ()

public static MatMul.Options transposeA (transposeA booléenne)

Paramètres
transposerA Si c'est vrai, "a" est transposé avant la multiplication.

public statique MatMul.Options transposeB (transposeB booléenne)

Paramètres
transposerB Si c'est vrai, "b" est transposé avant la multiplication.