I want to know how to use sigmoid output.
This is what I thought.
Anyone can check this?
Example
Sigmoid output:
tensor([[0.4361],
        [0.4571],
        [0.4319],
        [0.4922],
        [0.5832],
        [0.5590],
         ...

if output >= 0.5:
   result = '0'
else:
   result = '1'