torch.nn.functional.unfold
- torch.nn.functional.unfold(input, kernel_size, dilation=1, padding=0, stride=1)[source][source]
从批处理的输入张量中提取滑动局部块。
警告
目前仅支持 4-D 输入张量（批量的图像类张量）。
警告
展开后的张量中可能多个元素指向同一个内存位置。因此，原地操作（尤其是向量化操作）可能会导致不正确的行为。如果您需要向张量写入，请先对其进行克隆。
详细信息请见
torch.nn.Unfold。- 返回类型:
