Skip to content

combinatorial_probabilistic_compression

Bases: combinatorial_compression

Source code in tinybig/compression/combinatorial_compression.py 
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
class combinatorial_probabilistic_compression(combinatorial_compression):
    def __init__(
        self,
        name: str = 'combinatorial_probabilistic_compression',
        d: int = 1, k: int = 1,
        metric: Callable[[torch.Tensor], torch.Tensor] = None,
        with_replacement: bool = False,
        require_normalization: bool = True,
        *args, **kwargs
    ):
        super().__init__(
            name=name,
            d=d, k=k,
            metric=metric,
            simply_sampling=False,
            log_prob=True,
            with_replacement=with_replacement,
            require_normalization=require_normalization,
            *args, **kwargs
        )