0
Je dois binariser la sortie o d'un modèle de torche (script lua), la plage de valeurs est [-1, + 1], je veux seuil de ces valeurs dans un tel manière:comment effectuer la binarisation de sortie d'un modèle de torche
0 if o[i]<0
1 if o[i]>=0
La sortie est composé de 32 couches avec tenseurs flotteur taille 1x1, donc 32 flotteurs, je veux obtenir 32 bits de ces 32 flotteurs, mais je ne peux pas trouver une couche qui permet de le faire. À l'heure actuelle, j'ai un cycle pour vérifier la valeur de chaque niveau, mais il est très lent.
Peut-être que je peux utiliser la couche de seuil ou mettre en œuvre un par moi-même, avez-vous des conseils?