BCEWITHLOGITSLOSS binary classification

You could try looking at this post. You have to have the same number of weights as classes.