The output and criterion look alright for a multi-label classification.
Make sure that the target has the same shape as output.
Are you seeing any issues, e.g. lower accuracy etc.?
PS: Variables are deprecated and you can just use tensors