mindspore.mint.nn.Unfold
- class mindspore.mint.nn.Unfold(kernel_size, dilation=1, padding=0, stride=1)[source]
Extracts sliding local blocks from a batched input tensor.
For details, please refer to
mindspore.mint.nn.functional.unfold()
.- Supported Platforms:
Ascend
Examples
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, mint >>> input = Tensor(np.random.rand(4, 4, 32, 32), mindspore.float64) >>> unfold = mint.nn.Unfold(kernel_size=3, dilation=1, stride=1) >>> output = unfold(input) >>> print(output.shape) (4, 36, 900)