Go it. Thanks! We are probably not using the same versions as I did not compile pytorch from scratch and so I have the following function signature:
torch.nn.functional.binary_cross_entropy_with_logits(input, target, weight=None, size_average=True)
Thanks anyway