pytorch variable out[out > 0] = 1

发表于 | 分类于 | 阅读次数: 
1
2
3
4
5
# out[out > 0] = 1
# out[out <= 0] = -1
out = (out >= 0.5).float()
# map 1,0 to 1,-1
out = out * 2 - 1
请作者喝一杯咖啡☕️