torch.nn.functional.unfold

torch.nn.functional.unfold(input, kernel_size, dilation=1, padding=0, stride=1)

從批次輸入張量中提取滑動區域性塊。

警告

目前，僅支援 4-D 輸入張量（批次影像狀張量）。

警告

展開後的張量中的多個元素可能指向同一個記憶體位置。因此，就地操作（尤其是向量化操作）可能導致不正確的行為。如果您需要寫入張量，請先克隆它。

詳見 torch.nn.Unfold

返回型別

Tensor