torch.unflod与torch.nn.unfold用法

PyTorch 中的两个函数：torch.unfold 和 torch.nn.unfold。它们分别用于不同的目的，让我们分别来理解一下：

1. torch.nn.Unfold 类

• 功能： 类似于函数 torch.unfold，torch.nn.Unfold 类也用于沿着指定维度滑动提取窗口并将每个窗口展平。与函数不同的是，torch.nn.Unfold 是一个可学习的层，可以作为神经网络的一部分进行训练。

• 定义：

  torch.nn.Unfold(kernel_size, dilation=1, padding=0, stride=1)
  

• 参数：

  • kernel_size (int or tuple): 窗口的大小。
  • dilation (int or tuple, optional): 卷积核元素之间的间距，默认为 1。
  • padding (int or tuple, optional): 填充的大小，默认为 0。
  • stride (int or tuple, optional): 窗口滑动的步长，默认为 1。

• 使用方法：

   import torch
   from torch.nn import Unfold
   
   # 定义 Unfold 层
   unfold = Unfold(kernel_size=(2, 2),dilation=1, padding=0, stride=(1, 1))
   
   # 输入张量 (注意数据类型转换)
   x = torch.arange