avg_pool3d

    该函数是一个三维平均池化函数, 根据输入参数 kernel_size, stride, padding 等参数对输入做平均池化操作。

    例如:

    输入:

    X 形状:

    属性:

    kernel_size:

    stride:

    avg_pool3d - 图1

    输出:

    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 一致。