pytorch中函数名后有下划线和没有下划线(e.g clamp和clamp_)的区别

一般来说,在pytorch中如果对tensor的一个函数后加上了下划线,则表明这是一个in-place类型,所谓in-place类型是指在一个tensor上操作了之后,是直接修改了这个tensor,还是返回一个新的tensor,而旧的tensor并不修改。

例如:

clamp和clamp_

可以看到,函数名后加下划线会修改原始输入,即直接改了这个tensor。

发布了15 篇原创文章 · 获赞 19 · 访问量 8806

猜你喜欢

转载自blog.csdn.net/weixin_42448226/article/details/104824239