Class UnfoldImpl¶
Defined in File fold.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::Cloneable< UnfoldImpl >
(Template Class Cloneable)
Class Documentation¶
-
class UnfoldImpl : public torch::nn::Cloneable<UnfoldImpl>¶
Applies unfold over a 4-D input.
See https://pytorch.org/docs/main/nn.html#torch.nn.Unfold to learn about the exact behavior of this module.
See the documentation for
torch::nn::UnfoldOptions
class to learn what constructor arguments are supported for this module.Example:
Unfold model(UnfoldOptions({2, 4}).dilation(2).padding({2, 1}).stride(2));
Public Functions
-
inline UnfoldImpl(ExpandingArray<2> kernel_size)¶
-
explicit UnfoldImpl(const UnfoldOptions &options_)¶
-
virtual void reset() override¶
reset()
must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.
-
virtual void pretty_print(std::ostream &stream) const override¶
Pretty prints the
Unfold
module into the givenstream
.
-
Tensor forward(const Tensor &input)¶
Public Members
-
UnfoldOptions options¶
The options with which this
Module
was constructed.
-
inline UnfoldImpl(ExpandingArray<2> kernel_size)¶