Python–PyTorch 夹具()方法
原文:https://www.geeksforgeeks.org/python-pytorch-clamp-method/
PyTorch torch.clamp()方法将所有输入元素夹在[ min，max ]范围内，并返回结果张量。
语法:
torch.clamp(inp, min, max, out=None)论据
- inp: 这是输入张量。
- min: 这是一个数字，指定输入要箝位到的范围的下限。
- 最大值:这是一个数字，指定输入要箝位到的范围的上限。
- 出:输出张量。
返回:它返回一个张量。
Let’s see this concept with the help of few examples:Example 1:
# Importing the PyTorch library
import torch
# A constant tensor of size n
a = torch.randn(6)
print(a)
# Applying the clamp function and
# storing the result in 'out'
out = torch.clamp(a, min = 0.5, max = 0.9)
print(out)
输出:
-0.9214
-0.1268
1.1570
-0.2753
-0.0746
0.7957
[torch.FloatTensor of size 6]
0.5000
0.5000
0.9000
0.5000
0.5000
0.7957
[torch.FloatTensor of size 6]
例 2:
# Importing the PyTorch library
import torch
# A constant tensor of size n
a = torch.FloatTensor([1, 4, 6, 8, 10, 14])
print(a)
# Applying the clamp function and
# storing the result in 'out'
out = torch.clamp(a, min = 5, max = 10)
print(out)
输出:
1
4
6
8
10
14
[torch.FloatTensor of size 6]
5
5
6
8
10
10
[torch.FloatTensor of size 6]?
