Closed f-dangel closed 3 years ago
This package provides three classes that inherit from torch.nn.Module:
torch.nn.Module
unfoldNd.Unfold1d
unfoldNd.Unfold2d
unfoldNd.Unfold3d
All three functions have the same interface as torch.nn.Unfold.
torch.nn.Unfold
Internally, they call out to a function _unfoldNd which uses a one-hot convolution to perform the unfold operation.
_unfoldNd
unfold
This package provides three classes that inherit from
torch.nn.Module
:unfoldNd.Unfold1d
unfoldNd.Unfold2d
unfoldNd.Unfold3d
All three functions have the same interface as
torch.nn.Unfold
.Internally, they call out to a function
_unfoldNd
which uses a one-hot convolution to perform theunfold
operation.