Add RandomRotation to docs (#362)

080b9543 · David de la Iglesia Castro · Alykhan Tejani · 6c2ac026 · 080b9543 · 080b9543
Commit 080b9543 authored Dec 07, 2017 by David de la Iglesia Castro Committed by Alykhan Tejani Dec 07, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

docs/source/transforms.rst docs/source/transforms.rst +2 -0

torchvision/transforms/functional.py torchvision/transforms/functional.py +1 -1

No files found.
--- a/docs/source/transforms.rst
+++ b/docs/source/transforms.rst
@@ -38,6 +38,8 @@ Transforms on PIL Image
 .. autoclass:: ColorJitter
+.. autoclass:: RandomRotation
 Transforms on torch.\*Tensor
 ----------------------------

--- a/torchvision/transforms/functional.py
+++ b/torchvision/transforms/functional.py
@@ -527,7 +527,7 @@ def adjust_gamma(img, gamma, gain=1):
 def rotate(img, angle, resample=False, expand=False, center=None):
-    """Rotate the image by angle and then (optionally) translate it by (n_columns, n_rows)
+    """Rotate the image by angle.
    Args: