Hi,
As understood from the topic, I want to implement cross entropy with softmax on logits with target of format [batchsize,C,H,W] where values are in the range [0,1). Values across axes 1 does not sum to 1.
torch.nn.functional. cross_entropy does not allow 4D targets.
How do I do this?