model.py 文件源码

def __call__(self, x):
        segs = list(itertools.accumulate(
            clf.n_input for clf in self.classifiers
        ))
        if segs:
            xs = cf.split_axis(x, segs, 1)
        else:
            xs = [x]

        y = self.segmenter(xs[-1])
        zs = tuple(clf(x) for x, clf in zip(xs[:-1], self.classifiers))
        return y, zs