2017-04-11 6 views
0

トーチモデル(ルアスクリプト)の出力oを2進化する必要があります。値の範囲は[-1、+ 1]です。する方法:出力はサイズ1x1のフロートテンソルなので、32台の山車で32層で構成されてトーチモデルの出力二進化のしくみ

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

は、私はそれらの32台のフロートから32ビットを取得したいが、私はそれを行うことを可能にする層を見つけることができません。

現時点では、各レベルの値をチェックするサイクルがありますが、非常に遅いです。

多分私は閾値レイヤーを使用することも、自分で実装することもできます。アドバイスはありますか?

答えて

関連する問題