模型开发接口
推理部署接口
API 相关说明
对滑动窗口对应的输入数据求和。
更多信息参见 SlidingWindowTranspose 。
SlidingWindowTranspose
inp (Tensor) – 输入tensor。
Tensor
output_size (Union[int, Tuple[int, int]]) – 输出tensor的shape。
Union
int
Tuple
kernel_size (Union[int, Tuple[int, int]]) – 窗口的尺寸。
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
上一页
megengine.functional.nn.sliding_window
下一页
megengine.functional.nn.avg_pool2d