Class FoldImpl¶
Defined in File fold.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::Cloneable< FoldImpl >
(Template Class Cloneable)
Class Documentation¶
-
class FoldImpl : public torch::nn::Cloneable<FoldImpl>¶
Applies fold over a 3-D input.
See https://pytorch.org/docs/main/nn.html#torch.nn.Fold to learn about the exact behavior of this module.
See the documentation for
torch::nn::FoldOptions
class to learn what constructor arguments are supported for this module.Example:
Fold model(FoldOptions({8, 8}, {3, 3}).dilation(2).padding({2, 1}).stride(2));
Public Functions
-
inline FoldImpl(ExpandingArray<2> output_size, ExpandingArray<2> kernel_size)¶
-
explicit FoldImpl(const FoldOptions &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
Fold
module into the givenstream
.
-
Tensor forward(const Tensor &input)¶
Public Members
-
FoldOptions options¶
The options with which this
Module
was constructed.
-
inline FoldImpl(ExpandingArray<2> output_size, ExpandingArray<2> kernel_size)¶