Two more missed as_subclass

Author: datumbox

torchvision/prototype/transforms/_geometry.py

@@ -676,10 +676,12 @@ def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
         output = F.crop(inpt, top=params["top"], left=params["left"], height=params["height"], width=params["width"])
 
         if isinstance(output, features.BoundingBox):
+            output = output.as_subclass(torch.Tensor)
             bboxes = output[is_within_crop_area]
             bboxes = F.clamp_bounding_box(bboxes, output.format, output.spatial_size)
             output = features.BoundingBox.wrap_like(output, bboxes)
         elif isinstance(output, features.Mask):
+            output = output.as_subclass(torch.Tensor)
             # apply is_within_crop_area if mask is one-hot encoded
             masks = output[is_within_crop_area]
             output = features.Mask.wrap_like(output, masks)