frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

ReshapeTransform: added missing argument in docstring (#144401)

Browse Source 
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>
This commit is contained in:
Randolf Scholz
2025-01-13 17:59:56 +00:00
committed by PyTorch MergeBot
parent fe8c5c7a2d
commit 983bf604e5
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/distributions/transforms.py
View File

@ -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