用张量对张量切片

Question

我想做一个移动窗口的张量。我正在使用列表推导式，但是，这是顺序的，使得它非常慢。

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)

我见过"索引"的方法。但不是切片。有办法吗?

Answer 1

似乎你正在寻找内核大小为(1, self.l_c)的w展开。