:github_url: https://github.com/pytorch/pytorch


.. _program_listing_file_torch_csrc_api_include_torch_nn_modules_fold.h:

Program Listing for File fold.h
===============================

|exhale_lsh| :ref:`Return to documentation for file <file_torch_csrc_api_include_torch_nn_modules_fold.h>` (``torch/csrc/api/include/torch/nn/modules/fold.h``)

.. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS

.. code-block:: cpp

   #pragma once
   
   #include <torch/expanding_array.h>
   #include <torch/nn/cloneable.h>
   #include <torch/nn/functional/fold.h>
   #include <torch/nn/options/fold.h>
   #include <torch/nn/pimpl.h>
   #include <torch/types.h>
   
   namespace torch::nn {
   
   class TORCH_API FoldImpl : public torch::nn::Cloneable<FoldImpl> {
    public:
     FoldImpl(ExpandingArray<2> output_size, ExpandingArray<2> kernel_size)
         : FoldImpl(FoldOptions(output_size, kernel_size)) {}
     explicit FoldImpl(const FoldOptions& options_);
   
     void reset() override;
   
     void pretty_print(std::ostream& stream) const override;
   
     Tensor forward(const Tensor& input);
   
     FoldOptions options;
   };
   
   TORCH_MODULE(Fold);
   
   // ============================================================================
   
   class TORCH_API UnfoldImpl : public Cloneable<UnfoldImpl> {
    public:
     UnfoldImpl(ExpandingArray<2> kernel_size)
         : UnfoldImpl(UnfoldOptions(kernel_size)) {}
     explicit UnfoldImpl(const UnfoldOptions& options_);
   
     void reset() override;
   
     void pretty_print(std::ostream& stream) const override;
   
     Tensor forward(const Tensor& input);
   
     UnfoldOptions options;
   };
   
   TORCH_MODULE(Unfold);
   
   } // namespace torch::nn