fn
avg_pool2d_op
→TensorImplPtrint avg_pool2d_op(const int & x, int KH, int KW, int stride_h, int stride_w, int pad_h, int pad_w)Python wrappers
Public Python APIs implemented by this engine symbol.Two-dimensional average pooling over a batch of feature maps.
Smooth, differentiable counterpart to max_pool2d_op. A common
choice for global average pooling heads (use a kernel equal to the
full spatial extent).
Parameters
xTensorImplPtrInput of shape
(B, C, H, W).KH, KWintPer-axis window size.
stride_h, stride_wint= NonePer-axis stride.
0 is a sentinel for non-overlapping windows. Default: 0.pad_h, pad_wint= NonePer-axis zero-padding. Default:
0.Returns
TensorImplPtrOutput of shape (B, C, H_out, W_out).