nn.AdaptiveAvgPool3d

nn.AdaptiveAvgPool3d
官方解释：

class AdaptiveAvgPool3d(_AdaptiveAvgPoolNd)
 |  AdaptiveAvgPool3d(output_size: Union[int, Tuple[int, ...]]) -> None
 |  
 |  Applies a 3D adaptive average pooling over an input signal composed of several input planes.
 |  
 |  The output is of size D x H x W, for any input size.
 |  The number of output features is equal to the number of input planes.
 |  
 |  Args:
 |      output_size: the target output size of the form D x H x W.
 |                   Can be a tuple (D, H, W) or a single number D for a cube D x D x D.
 |                   D, H and W can be either a ``int``, or ``None`` which means the size will
 |                   be the same as that of the input.

nn.AdaptiveAvgPool3d(output_size)
查看上面对output_size的解释
也就是说，我们只需要关注输出维度的大小 output_size ，具体的实现过程和参数选择自动帮你确定了。