torch.nn.functional.unfold
-
torch.nn.functional.unfold(input, kernel_size, dilation=1, padding=0, stride=1)[source] -
Извлекает скользящие локальные блоки из тензора входного массива.
Предупреждение
В настоящее время поддерживаются только тензоры входных данных 4-го порядка (массивы тензоров, похожие на изображение).
Предупреждение
Более одного элемента развернутого тензора может ссылаться на одно и то же место в памяти. В результате операции на месте (особенно векторизованные) могут привести к неправильному поведению. Если вам необходимо записать в тензор, сначала скопируйте его.
См.
torch.nn.Unfoldдля подробностей- Тип возвращаемого значения:
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.nn.functional.unfold.html