torch.clamp_()函數
1.clamp()函數:
參考資料:https://blog.csdn.net/jacke121/article/details/85270621
1)函數功能:
clamp()函數的功能將輸入input張量每個元素的值壓縮到區間 [min,max],並返回結果到一個新張量。
2)參數列表:
torch.clamp(input, min, max, out=None) → Tensor
input:輸入張量;
min:限制範圍下限;
max:限制範圍上限;
out:輸出張量。
3)應用舉例:
a=torch.randint(low=0,high=10,size=(10,1))
print(a)
b=torch.clamp(a,3,9)
print(b)
2.clamp_()與clamp()的區別:
參考資料:https://blog.csdn.net/qq_27261889/article/details/86777230
pytorch中,一般來說如果對tensor的一個函數後加上了下劃線,則表明這是一個in-place類型。in-place類型是指,當在一個tensor上操作了之後,是直接修改了這個tensor,而不是返回一個新的tensor並不修改舊的tensor。