Bases: object
MSE loss with top-k truncation: ||x - T^{-1}(truncate(T(x), k))||^2.
- Parameters:
k (int) – Number of coefficients to keep after top-k magnitude truncation.
Must be positive.
-
__init__(k)
- Parameters:
k (int)
- Return type:
None
Methods
Attributes
-
k: int
