torch实现tensor二维数据的滑动窗口

初始化一个大小为n*m的tensor,我这里用的是4*25

x = torch.tensor.arange(0,100).reshpe(3,4)

滑动时间窗口大小设置为w,步长设置为s,滑动次数为i,可以计算出窗口右侧到达的位置p,大概可以得到相应的公式，不一定刚好滑到边缘，主要是方便计算下滑动的次数。

我用的窗口大小为4，步长是4，滑动次数为6，4+4(6-1) = 24,停留在第24列，25列滑动窗口不选取。

#  unfold(dimension, size, step) -> Tensor
# 第一个参数代表维度，方便自己理解我是从每行开始滑动，
x = x.unfold(1,4,4)
print(x.shape) #输出torch.Size([4, 6, 4])

得到如下的tensor

我需要按列获取窗口数据，使用transpose对维度进行转换。

x = x.transpose(0,1)
print(x.shape) #输出torch.Size([6, 4, 4])

最终效果：

原始数据

处理之后：