torch.clamp()函數用于將輸入張量中的元素限制在指定的范圍內。
函數的語法如下:
torch.clamp(input, min, max, out=None)
參數說明:
使用示例:
import torch
x = torch.randn(5) # 創建一個包含5個隨機數的張量
print(x) # 打印原始張量
# 使用clamp函數將張量的元素限制在-0.5到0.5之間
y = torch.clamp(x, -0.5, 0.5)
print(y) # 打印限制后的張量
輸出結果示例:
tensor([ 0.0849, -0.2706, 0.7244, 0.0921, 0.6237])
tensor([ 0.0849, -0.2706, 0.5000, 0.0921, 0.5000])
在上面的示例中,首先創建了一個包含5個隨機數的張量x,然后使用clamp函數將該張量的元素限制在-0.5到0.5之間,結果存儲在張量y中。最后打印出原始張量和限制后的張量。可以看到,張量y中的元素都被限制在了-0.5到0.5之間。