ReshapeTransform: added missing argument in docstring (pytorch#144401)

See pytorch#144197 (comment) Pull Request resolved: pytorch#144401 Approved by: https://github.com/janeyx99, https://github.com/malfet Co-authored-by: Nikita Shulga <[email protected]> Co-authored-by: Jane (Yuan) Xu <[email protected]>
jagadish-amd · Jan 13, 2025 · 983bf60 · 983bf60
1 parent fe8c5c7
commit 983bf60
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch/distributions/transforms.py b/torch/distributions/transforms.py
@@ -480,6 +480,8 @@ class ReshapeTransform(Transform):
     Arguments:
         in_shape (torch.Size): The input event shape.
         out_shape (torch.Size): The output event shape.
+        cache_size (int): Size of cache. If zero, no caching is done. If one,
+            the latest single value is cached. Only 0 and 1 are supported. (Default 0.)
     """
 
     bijective = True