avg_pool3d
该函数是一个三维平均池化函数, 根据输入参数 kernel_size, stride, padding 等参数对输入做平均池化操作。
例如:
输入:
X 形状:
属性:
kernel_size:
stride:
输出:
Out 形状:(N,C,Din,Hout,Wout)(N,C,Din,Hout,Wout)
out(Ni,Cj,d,h,w)\=∑k\=0kD−1∑m\=0kH−1∑n\=0kW−1input(Ni,Cj,stride[0]×d+k,stride[1]×h+m,stride[2]×w+n)kD×kH×kWout(Ni,Cj,d,h,w)\=∑k\=0kD−1∑m\=0kH−1∑n\=0kW−1input(Ni,Cj,stride[0]×d+k,stride[1]×h+m,stride[2]×w+n)kD×kH×kW
5-D Tensor,数据类型与输入 x 一致。