pytorch 方法名后加下划线,如 clamp() 与 clamp_()

任何可以改变tensor内容的操作都会在方法名后加一个下划线'_'
例如:x.copy_(y), x.t_(), 这俩都会改变x的值

在pytorch中,我们需要注意,对于tensor的操作时候是in-place类型。
in-place类型是指,但在一个tensor上操作了之后,是直接修改了这个tensor。

非in-place类型,是指返回一个新的tensor,而旧的tensor并不修改。

pytorch中,一般来说,如果对tensor的一个函数后加上了下划线,则表明这是一个in-place类型旧的tensor也修改了,
如:
clamp 和 clamp_ (以下为实际例子)
 

 å¨è¿éæå¥å¾çæè¿°

发布了72 篇原创文章 · 获赞 33 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/Elvirangel/article/details/104333715