共計 520 個字符,預計需要花費 2 分鐘才能閱讀完成。
torch.clamp() 函數用于將輸入張量中的元素限制在指定的范圍內。
函數的語法如下:
torch.clamp(input, min, max, out=None)
參數說明:
- input:輸入的張量。
- min:最小值,小于該值的元素將被替換為該值。
- max:最大值,大于該值的元素將被替換為該值。
- out:用于存儲輸出結果的張量。
使用示例:
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 之間。
丸趣 TV 網 – 提供最優質的資源集合!
正文完