The printed shapes correspond to the shape mismatch mentioned in the error message.
How would you like to compute the binary cross entropy with a different number of samples?
Did you change anything in the example code?
I just retried it with the latest nightly build and it seems to work without any errors.