topk#
Module Contents#
Compressor for federated communication |
- class TopkCompressor(compress_ratio)#
Bases:
fedlab.contrib.compressor.compressor.Compressor
Compressor for federated communication Top-k gradient or weights selection :param compress_ratio: compress ratio :type compress_ratio: float
- compress(tensor)#
compress tensor into (values, indices) :param tensor: tensor :type tensor: torch.Tensor
- 返回
(values, indices)
- 返回类型
- decompress(values, indices, shape)#
decompress tensor