Hi!
So the following pdb output states the question very clearly:
(Pdb) type(labels.cuda())
<class ‘torch.cuda.FloatTensor’>
(Pdb) type(predicted.data)
<class ‘torch.cuda.FloatTensor’>
(Pdb) predicted.data>=predicted.data
1
1
1
1
[torch.cuda.ByteTensor of size 4x1 (GPU 0)]
(Pdb) predicted.data>=labels.cuda()
0  0  0  0
0  0  0  0
0  0  0  0
0  0  0  0
[torch.cuda.ByteTensor of size 4x4 (GPU 0)]
So yeah, I am confused when it comes to the last output,is that intentional or a bug?