Fix docstring for heatmaps_to_keypoints

[raimbekovm]

Summary

Fixes #7148

The docstring for heatmaps_to_keypoints was incorrect:

• Stated output shape as (#rois, 4, #keypoints) but actual shape is (#rois, #keypoints, 3)
• Stated last dimension contains (x, y, logit, prob) but actually contains (x, y, visibility)
• Did not mention that the function returns a tuple of two tensors

Changes

Updated docstring to follow Google style with accurate Args and Returns sections that match the actual function behavior:

• xy_preds: Tensor[K, N, 3] containing (x, y, v) coordinates and visibility
• scores: Tensor[K, N] containing heatmap scores at predicted locations

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/9323

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit e630850 with merge base 1e53952 ():
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.