一般来说,在pytorch中如果对tensor的一个函数后加上了下划线,则表明这是一个in-place类型,所谓in-place类型是指在一个tensor上操作了之后,是直接修改了这个tensor,还是返回一个新的tensor,而旧的tensor并不修改。
例如:
clamp和clamp_
可以看到,函数名后加下划线会修改原始输入,即直接改了这个tensor。
2024-01-07 15:48:04
一般来说,在pytorch中如果对tensor的一个函数后加上了下划线,则表明这是一个in-place类型,所谓in-place类型是指在一个tensor上操作了之后,是直接修改了这个tensor,还是返回一个新的tensor,而旧的tensor并不修改。
例如:
clamp和clamp_
可以看到,函数名后加下划线会修改原始输入,即直接改了这个tensor。