0
我必須對火炬模型（lua腳本）的輸出o進行二進制化，值範圍爲[-1，+ 1]，我希望將這些值限制在這樣的值這種方式：如何執行火炬模型的輸出二值化
0 if o[i]<0
1 if o[i]>=0
的輸出由32層用大小1x1的浮張量，所以32個浮標組成，我想從這些32個浮標得到32位，但我不能找到一個層，它允許這樣做。
目前我有一個for循環檢查每個級別的值，但它非常緩慢。
也許我可以使用閾值層或由我自己實現一個，你有什麼建議嗎？