rotated bounding box pytorch