QuantizedMatMulWithBias

classe finale publique QuantizedMatMulWithBias

Effectue une multiplication matricielle quantifiée de « a » par la matrice « b » avec ajout de biais.

Les entrées doivent être des matrices bidimensionnelles et un vecteur de biais 1D. Et la dimension intérieure de « a » (après avoir été transposée si « transpose_a » est non nul) doit correspondre à la dimension extérieure de « b » (après avoir été transposée si « transposed_b » est non nul). Ensuite, effectuez une opération d'ajout de diffusion avec des valeurs de biais sur le résultat de la multiplication matricielle. La taille du biais doit correspondre à la dimension intérieure de « b ».

Classes imbriquées

classe QuantizedMatMulWithBias.Options Attributs facultatifs pour QuantizedMatMulWithBias

Méthodes publiques

statique <W, T, U, V> QuantizedMatMulWithBias <W>
créer ( Portée portée, Opérande <T> a, Opérande <U> b, Opérande <V> biais, Opérande <Float> minA, Opérande <Float> maxA, Opérande <Float> minB, Opérande <Float> maxB, Classe<W > Sortie, Options... (options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération QuantizedMatMulWithBias.
statique QuantizedMatMulWithBias.Options
inputQuantMode (chaîne inputQuantMode)
Sortie <Flottant>
sortie max ()
La valeur flottante que représente la valeur de sortie quantifiée la plus élevée.
Sortie <Flottant>
minSortie ()
La valeur flottante que représente la valeur de sortie quantifiée la plus basse.
Sortie <W>
dehors ()
statique QuantizedMatMulWithBias.Options
transposeA (transposeA booléenne)
statique QuantizedMatMulWithBias.Options
transposeB (transposeB booléenne)

Méthodes héritées

Méthodes publiques

public static QuantizedMatMulWithBias <W> créer ( Portée de la portée , Opérande <T> a, Opérande <U> b, Biais de l' Opérande <V>, Opérande <Float> minA, Opérande <Float> maxA, Opérande <Float> minB, Opérande <Float > maxB, Classe<W> Toutput, Options... options)

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

Paramètres
portée portée actuelle
un Une matrice à multiplier. Doit être un tenseur bidimensionnel de type « quint8 ».
b Une matrice à multiplier et doit être un tenseur bidimensionnel de type « qint8 ».
biais Un tenseur de biais 1D avec une dimension intérieure correspondant à la taille de « b » (après avoir été transposé si « transposed_b » est non nul).
minA La valeur flottante que représente la valeur « a » quantifiée la plus basse.
maxA La valeur flottante que représente la valeur « a » quantifiée la plus élevée.
minB La valeur flottante que représente la valeur « b » quantifiée la plus basse.
maxB La valeur flottante que représente la valeur « b » quantifiée la plus élevée.
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de QuantizedMatMulWithBias

public statique QuantizedMatMulWithBias.Options inputQuantMode (String inputQuantMode)

Paramètres
inputQuantMode Mode de quantification des données d’entrée. Soit MIN_FIRST (par défaut) soit SCALED.

sortie publique <Float> maxOut ()

La valeur flottante que représente la valeur de sortie quantifiée la plus élevée.

sortie publique <Float> minOut ()

La valeur flottante que représente la valeur de sortie quantifiée la plus basse.

Sortie publique <W> out ()

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

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

public static QuantizedMatMulWithBias.Options transposeB (booléen transposeB)

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