OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
381d0cb239
pytorch/torch/nn
History
zeshengzong 1cc5f6b623 Optimize MaxPool1d param ceil_mode description (#148869) 
Fixes #148123

Add output shape formula based on `ceil_mode` value, according to

00199acdb8/aten/src/ATen/native/Pool.h (L61-L75)

## Test Result

### Before

![image](https://github.com/user-attachments/assets/0a175178-a104-4348-a14b-516e866d533a)

### After

![image](https://github.com/user-attachments/assets/ce621d4b-1986-41fb-bd71-2b03c0aa996e)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/148869
Approved by: https://github.com/mikaylagawarecki
2025-03-17 08:50:40 +00:00
..
attention [SDPA] Respect sdpa_kernel's priority_order setting in torch.compile (#147768) 2025-03-13 18:52:34 +00:00
backends
intrinsic
modules Optimize MaxPool1d param ceil_mode description (#148869) 2025-03-17 08:50:40 +00:00
parallel [BE][Ez]: Use itertools.chain.from_iterable when possible (#148190) 2025-03-06 20:37:06 +00:00
qat
quantizable
quantized
utils Add hint message for pack_padded_sequence (#146747) 2025-02-20 06:27:07 +00:00
__init__.py
_reduction.py
common_types.py [5/N] Apply Ruff fixes and pyupgrade to Python 3.9 (#144205) 2025-01-15 04:00:47 +00:00
cpp.py
functional.py Fix torch.nn.functional.one_hot param num_classes optional description (#146470) 2025-02-06 07:48:05 +00:00
functional.pyi.in Use Python 3.9 typing (#148157) 2025-03-04 03:09:55 +00:00
grad.py
init.py
parameter.py
parameter.pyi remove allow-untyped-defs from torch/nn/parameter.pyi (#144654) 2025-01-13 19:02:31 +00:00