One possible approach would be to set the lb_ignore to nb_classes+1, create an approprialty shaped zero tensor, and remove this additional dimension using slicing.
Probably not the best approach, but it should work.