Closed f-dangel closed 3 years ago
Resolves #2, but proposes a slightly more compact API.
There are two functions:
unfoldNd.UnfoldNd
torch.nn.Unfold
unfoldNd.unfoldNd
torch.nn.functional.unfold
Cleans up some dummy functions and tests from the python library template.
Turns off darglint and pydocstyle as the docstrings link to the PyTorch implementations.
darglint
pydocstyle
Resolves #2, but proposes a slightly more compact API.
There are two functions:
unfoldNd.UnfoldNd
: Module class that can be used as drop-in replacement fortorch.nn.Unfold
, but accepts 3d, 4d, and 5d inputs.unfoldNd.unfoldNd
: Functional that can be used as drop-in replacement fortorch.nn.functional.unfold
, but accepts 3d, 4d, and 5d inputs.Cleans up some dummy functions and tests from the python library template.
Turns off
darglint
andpydocstyle
as the docstrings link to the PyTorch implementations.