我必须对火炬模型（lua脚本）的输出o进行二值化，值范围为[-1，+ 1]，我想以这样的方式对这些值进行阈值处理：
0 if o[i]<0
1 if o[i]>=0
输出由32层组成，大小为1x1浮点张量，所以32个浮点数，我想从这32个浮点数得到32位但我找不到允许这样做的层。
目前我有一个for循环来检查每个级别的值，但它非常慢。
也许我可以使用阈值层或自己实现一个，你有什么建议吗？
答案 0 :(得分：1)
您可以使用'大于或等于'运算符https://github.com/torch/torch7/blob/master/doc/maths.md#torchgea-b
local threshold_tensor = o:ge(0)