Solved by ptrblck in post #2 
              
              
                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 …