def test_3d_avg_shape(self):
        input_param = {'func': 'AVG',
                       'kernel_size': [3, 3, 2],
                       'stride': [3, 2, 1]}
        self._test_nd_downsample_output_shape(rank=3,
                                              param_dict=input_param,
                                              output_shape=(2, 6, 8, 16, 8))