如何沿特定维度堆叠火炬张量

Question

我有一个尺寸为 (x, H, W) 的火炬张量图像 我想每次都取一个相同大小的补丁（大小为 (x, h, w)，我总共有 k 个补丁

如何使用 for 循环沿 0 维堆叠我创建的所有补丁以获得大小为 (k,x,h,w) 的大补丁？我真的对火炬张量堆栈感到困惑。