def setup(self):
        splits = [0] + self.splits + [self.x.shape[0]]
        self.slices = []
        for i in range(len(splits) - 1):
            self.slices.append((splits[i], splits[i+1]))
        for i, (start, end) in enumerate(self.slices):
            shape = (end-start,) + self.x.shape[1:]
            self.outputs[i].shape = shape
            self.outputs[i].array = self.x.array[start:end]
            if self.bpropable:
                self.outputs[i].grad_array = ca.zeros(shape)
dataset/ETHPy150Open andersbll/deeppy/deeppy/expr/array.py/Slices.setup/original