其他分享
首页 > 其他分享> > pytorch中函数名后有下划线和没有下划线(e.g clamp和clamp_)的区别

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

作者:互联网

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

例如:

clamp和clamp_

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

标签:clamp,下划线,名后,pytorch,place,tensor
来源: https://blog.csdn.net/weixin_42448226/article/details/104824239