Sous-classes directes connues |
Normalise une TensorBuffer
avec une moyenne et un stddev donnés: sortie = (entrée - moyenne) / stddev.
Constructeurs publics
NormalizeOp(moyenne à virgule flottante, stddev flottant)
Initialise un NormalizeOp.
|
|
NormalizeOp(float[] moyenne, float[] stddev)
Initialise un NormalizeOp.
|
Méthodes publiques
TensorBuffer |
apply(entrée TensorBuffer)
Applique la normalisation définie sur un Tensor donné et renvoie le résultat.
|
Méthodes héritées
Constructeurs publics
public NormalizeOp (moyenne à virgule flottante, stddev)
Initialise un NormalizeOp. Lors de son appel, il crée un objet TensorBuffer
qui répond aux exigences suivantes :
output = (input - mean) / stddev
Dans les deux cas suivants, réinitialisez mean
sur 0 et stddev
sur 1 pour contourner la normalisation.
1. mean
et {code stddev} sont tous deux définis sur 0.
2. mean
est égal à 0 et {stddev} correspond à Infinity.
Remarque: Si mean
est défini sur 0 et que stddev
est défini sur 1, aucun calcul n'est effectué, et l'entrée d'origine est directement renvoyée lors de l'exécution.
Remarque: Actuellement, la valeur TensorBuffer
renvoyée est toujours un Tensor DataType.FLOAT32
, sauf que lorsque l'entrée est un Tensor DataType.UINT8
, mean
est défini sur 0 et stddev
est défini sur 1, de sorte que le Tensor DataType.UINT8
d'origine est renvoyé.
Paramètres
mean | la valeur moyenne à soustraire en premier. |
---|---|
stddev | la valeur de l'écart-type à diviser. |
Génère
IllegalArgumentException | Si stddev est égal à zéro.
|
---|
public NormalizeOp (float[] mean, float[] stddev)
Initialise un NormalizeOp. Lors de son appel, il crée un objet TensorBuffer
qui répond aux exigences suivantes :
// Pseudo code. [...][i] means a certain element whose channel id is i. output[...][i] = (input[...][i] - mean[i]) / stddev[i]
Remarque: Si toutes les valeurs dans mean
sont définies sur 0 et que toutes les stddev
sont définies sur 1, aucun calcul n'est effectué et l'entrée d'origine est directement renvoyée lors de l'exécution.
Remarque: Actuellement, la valeur TensorBuffer
renvoyée est toujours un Tensor DataType.FLOAT32
, sauf que l'entrée est un Tensor DataType.UINT8
. Toutes les valeurs mean
sont définies sur 0 et toutes les stddev
sont définies sur 1.
Paramètres
mean | les valeurs moyennes à soustraire en premier pour chaque canal. |
---|---|
stddev | les valeurs de l'écart type pour diviser ensuite pour chaque canal. |
Génère
IllegalArgumentException | si une valeur de stddev est nulle, si mean comporte un nombre différent d'éléments avec stddev ou si l'un d'entre eux est vide.
|
---|
Méthodes publiques
public TensorBuffer apply (TensorBuffer)
Applique la normalisation définie sur un Tensor donné et renvoie le résultat.
Remarque: input
est peut-être la même instance que la sortie.
Paramètres
entrée | du Tensor d'entrée. Il peut s'agir de la même instance que la sortie. |
---|
Renvoie
- du Tensor de sortie.