Method flatten

monai/transforms/compose.py:342–368

Return a Composition with a flattened list of transforms. Nested ``Compose`` objects that share the same ``map_items`` setting as the parent are inlined. Nested ``Compose`` objects with a *different* ``map_items`` value are kept as-is so their item-mapping behaviour is

(self)

        return None

    def flatten(self):
        """Return a Composition with a flattened list of transforms.

        Nested ``Compose`` objects that share the same ``map_items`` setting as
        the parent are inlined.  Nested ``Compose`` objects with a *different*
        ``map_items`` value are kept as-is so their item-mapping behaviour is
        preserved at runtime and during inversion.

        e.g., `t1 = Compose([x, x, x, x, Compose([Compose([x, x]), x, x])]).flatten()`
        will result in the equivalent of `t1 = Compose([x, x, x, x, x, x, x, x])`.

        """
        new_transforms = []
        for t in self.transforms:
            if type(t) is Compose and t.map_items == self.map_items:
                new_transforms += t.flatten().transforms
            else:
                new_transforms.append(t)

        return Compose(
            new_transforms,
            map_items=self.map_items,
            unpack_items=self.unpack_items,
            log_stats=self.log_stats,
            lazy=self._lazy,
            overrides=self.overrides,
        )

    def __len__(self):
        """Return number of transformations."""