networks.py 文件源码

def _make_overlay(self, location):
        if np.isscalar(location): location = [location]
        overlay = np.zeros(self.shapes[2], np.bool)
        for loc in location:
            overlay[:, :, loc, ...] = True
        return overlay