torch.nn.functional.unfold

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

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

警告

目前，僅支援 4 維輸入張量（批處理的類影像張量）。

警告

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

有關詳細資訊，請參閱 torch.nn.Unfold

返回型別

張量