pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 13:44:15 +08:00

Files

Rishub Tamirisa f3b8638074 Adding nn.ZeroPad1d and nn.ZeroPad3d (#96295 )

Fixes #95796

### Implementation
Adds python implementation for `nn.ZeroPad1d` and `nn.ZeroPad3d` in `torch/nn/modules/padding.py`.

Adds cpp implementation for `nn::ZeroPad1d` and `nn::ZeroPad3d` in the following 3 files, refactored with templates similarly to `nn::ConstantPad`'s implementation: <br>
- `torch/crsc/api/include/torch/nn/modules/padding.h`
- `torch/csrc/api/include/torch/nn/options/padding.h`
- `torch/csrc/api/src/nn/modules/padding.cpp`

Also added relevant definitions in `torch/nn/modules/__init__.py`.
### Testing
Adds the following tests:
-  cpp tests of similar length and structure as `ConstantPad` and the existing `ZeroPad2d` impl in `test/cpp/api/modules.cpp`
- cpp API parity tests in `torch/testing/_internal/common_nn.py`
- module init tests in `test/test_module_init.py`

Also added relevant definitions in `test/cpp_api_parity/parity-tracker.md`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/96295
Approved by: https://github.com/soulitzer

2023-03-10 03:51:41 +00:00

api

Adding nn.ZeroPad1d and nn.ZeroPad3d (#96295 )

2023-03-10 03:51:41 +00:00

c10d

Allow Process Group to support multiple backends (#88330 ) (#90997 )

2022-12-16 23:15:00 +00:00

common

Trim libshm deps, move tempfile.h to c10 (#17019 )

2019-02-13 19:38:35 -08:00

dist_autograd

set -Wsuggest-override for builds (#89852 )