megengine.functional.nn.sliding_window

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

从多batch输入tensor中提取滑窗大小的局部分块。

更多信息参见 SlidingWindow 。

参数

• inp (Tensor) – 输入tensor。

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

• padding (Union[int, Tuple[int, int]]) – 在两侧隐式添加零填充（zero padding）。默认：0

• stride (Union[int, Tuple[int, int]]) – window的步长。默认: 1

• dilation (Union[int, Tuple[int, int]]) – window的扩张。默认：1

返回类型

Tensor

上一页

megengine.functional.nn.deformable_conv2d

下一页

megengine.functional.nn.sliding_window_transpose