From 983bf604e5093703e22796bffb204f72da7e061c Mon Sep 17 00:00:00 2001 From: Randolf Scholz Date: Mon, 13 Jan 2025 17:59:56 +0000 Subject: [PATCH] ReshapeTransform: added missing argument in docstring (#144401) See https://github.com/pytorch/pytorch/pull/144197#discussion_r1907336339 Pull Request resolved: https://github.com/pytorch/pytorch/pull/144401 Approved by: https://github.com/janeyx99, https://github.com/malfet Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com> Co-authored-by: Jane (Yuan) Xu <31798555+janeyx99@users.noreply.github.com> --- torch/distributions/transforms.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torch/distributions/transforms.py b/torch/distributions/transforms.py index a1c912233f9665..8202f3a26a6af2 100644 --- 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