Pytorch Loss.data.item