Created by: heavengate
refix point_rcnn elementwise compatible For [2, 1] * [2] -> [2, 2] now, unsqueeze fg_mask in axis 1 to shape [-1, 1] to compatible to following elementwise operation
fg_mask