megengine.functional.nn.sliding_window

sliding_window(inp, kernel_size, padding=0, stride=1, dilation=1)

Extracts sliding local blocks from a batched input tensor.

Refer to SlidingWindow for more information.

参数

• inp (Tensor) – 输入张量。

• kernel_size (Union[int, Tuple[int, int]]) – 窗口的尺寸。

• padding (Union[int, Tuple[int, int]]) – implicit zero padding added on both sides of input. Default: 0

• stride (Union[int, Tuple[int, int]]) – stride of the window. Default: 1

• dilation (Union[int, Tuple[int, int]]) – dilation of the window. Default: 1

返回类型

Tensor

上一页

megengine.functional.nn.deformable_conv2d

下一页

megengine.functional.nn.sliding_window_transpose