melste:

tensor([[1.],
        [0.],
        [0.],
        [0.],
        ...
        [0.],
        [1.],
        [0.],
        [0.],
        [1.]], grad_fn=<SigmoidBackward>)



This is the value of which variable in your pipeline?