初始化一个大小为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])
最终效果:
原始数据
处理之后: