pytorch variable out[out > 0] = 1

# out[out > 0] = 1
# out[out <= 0] = -1
out = (out >= 0.5).float()
# map 1,0 to 1,-1
out = out * 2 - 1