Method register_spatial_cache

trellis/modules/sparse/basic.py:370–379

Register a spatial cache. The spatial cache can be any thing you want to cache. The registery and retrieval of the cache is based on current scale.

(self, key, value)

        return SparseTensor(feats=feats, coords=coords)

    def register_spatial_cache(self, key, value) -> None:
        """
        Register a spatial cache.
        The spatial cache can be any thing you want to cache.
        The registery and retrieval of the cache is based on current scale.
        """
        scale_key = str(self._scale)
        if scale_key not in self._spatial_cache:
            self._spatial_cache[scale_key] = {}
        self._spatial_cache[scale_key][key] = value

    def get_spatial_cache(self, key=None):
        """