That might be the case, as the docs 1 state that labels are expected to have the shape [bathc_size,].
I’m not familiar with the implementation and don’t know, if there is an easy switch to multi-label classification (couldn’t find the information).