torch.nn.functional.unfold

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

Извлекает скользящие локальные блоки из тензора входных данных в формате пакетной обработки.

Предупреждение

В настоящее время поддерживаются только тензоры входных данных размерностью 4 (тензоры, похожие на пакетные изображения).

Предупреждение

Более чем один элемент развернутого тензора может ссылаться на одно и то же место в памяти. В результате операции на месте (особенно векторизованные) могут привести к некорректному поведению. Если вам нужно записать в тензор, пожалуйста, сначала скопируйте его.

См. torch.nn.Unfold для подробностей

Тип возвращаемого значения

Тензор