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

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

例如:

clamp和clamp_

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

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

上一篇:css 计算值函数


下一篇:常用css模板