torch.clamp_()函數

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。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章