bartorch.hard_thresh

bartorch.hard_thresh()

Hard thresholding: keep x where |x| > lamda, zero elsewhere.

Returns:

The shape of input.

Return type:

torch.Tensor