今天看别人的代码,发现处理数据集使用了一个tensorly这个包里的unfold函数。很气的是我电脑上一直安装不上这个包。各种搜索,有说这个函数是滑动窗口,也有的说这个函数类似于切片。下面简单记录一下:
一般用法是:unfold(dim, size, step)
,表示作用的维度,一次选的元素个数和步长。
import torch
x = torch.Tensor([[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10],
[ 11, 12, 13, 14, 15],
[ 16, 17, 18, 19,