It seems that the F.binary_cross_entropy function takes two tensors a and b. It view both of them as vectors and compute the cross_entropy between a_i and b_i and then take sum of i.
I am wondering whether there is a way to compute a vector whose i-th coordinate is the binary cross entropy of a_i and b_i?