トーチモデル(ルアスクリプト)の出力oを2進化する必要があります。値の範囲は[-1、+ 1]です。する方法:出力はサイズ1x1のフロートテンソルなので、32台の山車で32層で構成されて 0 if o[i]<0
1 if o[i]>=0
は、私はそれらの32台のフロートから32ビットを取得したいが、私はそれを行うことを可能にする層を見つけることができません。 現時点では、各レベルの値をチェック
テンソル列を配列またはテンソルのインデックスで削除したいと思います。例: th> X = torch.rand(2,4)
th> X
0.7475 0.2512 0.6085 0.6414
0.7143 0.8299 0.2929 0.6945
[torch.DoubleTensor of size 2x4]
th> indices = torch.zeros(2)
th> in