wie die Ausgabe-Digitalisierung eines Brennermodells

Question

Ich binarisieren die Ausgabe o eines Fackel-Modells (lua Skript), der Wertebereich ist [-1, + 1], ich möchte Schwelle diese Werte in einem solchen Art und Weise, dass:

0 if o[i]<0 
1 if o[i]>=0 

Die Ausgabe von 32 Schichten mit einer Größe von 1x1 Schwimmer Tensoren zusammengesetzt ist, so 32 Schwimmern, ich will 32 Bits von den 32 Schwimmern bekommen, aber ich kann nicht eine Schicht, die ermöglicht, das zu tun finden.

Im Moment habe ich einen für Zyklus, der den Wert jedes Levels überprüft, aber es ist sehr langsam.

Vielleicht kann ich die Schwellenebene verwenden oder eine eigene implementieren, haben Sie einen Rat?

Answer 1

Sie können das 'größer oder gleich' Betreiber https://github.com/torch/torch7/blob/master/doc/maths.md#torchgea-b

local threshold_tensor = o:ge(0)