tf.nn.avg_pool3d