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

一般來說,在pytorch中如果對tensor的一個函數後加上了下劃線,則表明這是一個in-place類型,所謂in-place類型是指在一個tensor上操作了之後,是直接修改了這個tensor,還是返回一個新的tensor,而舊的tensor並不修改。 例如: clamp和clamp_ 可以看到,函數名後加下劃線會修改原始輸入,即直接改了這個tensor。
相關文章
相關標籤/搜索