:github_url: https://github.com/pytorch/pytorch .. _program_listing_file_torch_csrc_api_include_torch_nn_options_fold.h: Program Listing for File fold.h =============================== |exhale_lsh| :ref:`Return to documentation for file ` (``torch/csrc/api/include/torch/nn/options/fold.h``) .. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS .. code-block:: cpp #pragma once #include #include #include #include namespace torch::nn { struct TORCH_API FoldOptions { FoldOptions(ExpandingArray<2> output_size, ExpandingArray<2> kernel_size) : output_size_(output_size), kernel_size_(kernel_size) {} TORCH_ARG(ExpandingArray<2>, output_size); TORCH_ARG(ExpandingArray<2>, kernel_size); TORCH_ARG(ExpandingArray<2>, dilation) = 1; TORCH_ARG(ExpandingArray<2>, padding) = 0; TORCH_ARG(ExpandingArray<2>, stride) = 1; }; namespace functional { using FoldFuncOptions = FoldOptions; } // namespace functional // ============================================================================ struct TORCH_API UnfoldOptions { UnfoldOptions(ExpandingArray<2> kernel_size) : kernel_size_(kernel_size) {} TORCH_ARG(ExpandingArray<2>, kernel_size); TORCH_ARG(ExpandingArray<2>, dilation) = 1; TORCH_ARG(ExpandingArray<2>, padding) = 0; TORCH_ARG(ExpandingArray<2>, stride) = 1; }; namespace functional { using UnfoldFuncOptions = UnfoldOptions; } // namespace functional } // namespace torch::nn