support different class modes for bbox in box_with_nms_limit_op
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19820
Reviewed By: newstzpz
Differential Revision: D15112955
fbshipit-source-id: a757622a32cff7159c39735607103138dbbafc24