pytorch中函數名後有下劃線和沒有下劃線(e.g clamp和clamp_)的區別

一般來說,在pytorch中如果對tensor的一個函數後加上了下劃線,則表明這是一個in-place類型,所謂in-place類型是指在一個tensor上操作了之後,是直接修改了這個tensor,還是返回一個新的tensor,而舊的tensor並不修改。

例如:

clamp和clamp_

可以看到,函數名後加下劃線會修改原始輸入,即直接改了這個tensor。

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