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

例如:

clamp和clamp_

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

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

相关文章:

  • 2021-05-19
  • 2021-09-22
  • 2022-12-23
  • 2022-02-23
  • 2021-11-11
  • 2022-01-11
猜你喜欢
  • 2021-12-01
  • 2023-03-19
  • 2022-12-23
  • 2022-12-23
  • 2023-01-18
  • 2022-12-23
  • 2022-12-23
相关资源
相似解决方案