我想做一个移动窗口的Tensor，我使用的是列表解析，但是，这是连续的，所以速度非常慢。
weight_list = [w[:, :, :, i : i + self.l_c] for i in range(n)]
我想找
weight = torch.for_example.index_slice(w, start_indices=n, slice_length=self.l_c, dim=-1)
我见过"索引"的方法，但没有切片的方法。有没有这种方法？
mgdq6dx11#
您似乎正在寻找w的unfolding，内核大小为(1, self.l_c)。
w
(1, self.l_c)
1条答案
按热度按时间mgdq6dx11#
您似乎正在寻找
w的unfolding，内核大小为(1, self.l_c)。