torch.nn.functional.unfold

torch.nn.functional.unfold(input, kernel_size, dilation=1, padding=0, stride=1) [исходный код]

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

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

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

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

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

Подробности см. в torch.nn.Unfold

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

Tensor