bartorch.hard_thresh

bartorch.hard_thresh(lamda, input)

Hard thresholding: keep x where |x| > lamda, zero elsewhere (-H).

Returns:

The shape of input.

Return type:

torch.Tensor