functions.py 文件源码

def __init__(self, pyramid_height, pooling_class, use_cudnn=True):
        self._function = "spatial_pyramid_pooling_2d"
        self.pyramid_height = pyramid_height
        self.pooling_class = pooling_class
        self.use_cudnn = use_cudnn