如何执行火炬模型的输出二值化

Question

我必须对 Torch 模型（lua 脚本）的输出 o 进行二值化，值范围是 [-1,+1]，我想以这样的方式对这些值进行阈值处理：

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

输出由 32 个大小为 1x1 浮点张量的层组成，因此有 32 个浮点数，我想从这 32 个浮点数中获取 32 位，但我找不到允许这样做的层。

目前我有一个 for 循环来检查每个级别的值，但速度很慢。

也许我可以使用阈值层或自己实现一个，您有什么建议吗？

Answer 1

您可以使用“大于或等于”运算符https://github.com/torch/torch7/blob/master/doc/maths.md#torchgea-b

local threshold_tensor = o:ge(0)