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