Lua 如何执行火炬模型的输出二值化
我必须对torch模型（lua脚本）的输出进行二值化，其值范围为[-1，+1]，我希望以如下方式设置这些值的阈值：Lua 如何执行火炬模型的输出二值化,lua,torch,tensor,Lua,Torch,Tensor,我必须对torch模型（lua脚本）的输出进行二值化，其值范围为[-1，+1]，我希望以如下方式设置这些值的阈值： 0 if o[i]<0 1 if o[i]>=0 如果o[i]=0，则0 输出由32层组成，大小为1x1浮点张量，所以32个浮点，我想从这32个浮点中得到32位，但我找不到允许这样做的层 目前，我有一个for循环，它检查每个级别的值，但速度非常慢 也许我可以使用阈值层或自己实现一个阈值层，您有什么建议吗？您可以使用“大于或等于”操作符 local threshold
0 if o[i]<0
1 if o[i]>=0
0
也许我可以使用阈值层或自己实现一个阈值层，您有什么建议吗？您可以使用“大于或等于”操作符
local threshold_tensor = o:ge(0)