torch.nn.functional.threshold

torch.nn.functional.threshold(input, threshold, value, inplace=False)

Thresholds each element of the input Tensor.

See Threshold for more details.

Return type: Tensor

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.functional.threshold.html