torch.clamp
时间: 2023-10-12 07:21:12 浏览: 52
torch.clamp(input, min, max, out=None) -> Tensor
This function takes an input tensor and returns a new tensor with all the elements clamped to a specified range. The range is defined by the `min` and `max` values passed as arguments.
If an element in the input tensor is less than `min`, it is replaced with `min`. If it is greater than `max`, it is replaced with `max`. If the element is within the specified range, it is left unchanged.
The function can also take an optional output tensor `out` where the result is written. If `out` is not provided, a new tensor is returned.
Example:
```
import torch
x = torch.tensor([1, 2, 3, 4, 5])
y = torch.clamp(x, 2, 4)
print(y)
```
Output:
```
tensor([2, 2, 3, 4, 4])
```
In this example, we clamp the input tensor `x` to the range [2, 4], and store the result in `y`. The elements less than 2 are replaced with 2, elements greater than 4 are replaced with 4, and the rest are left unchanged.
阅读全文