Pytorch函数结尾下划线

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

转载自pytorch中函数名后有下划线和没有下划线 – cltt – 博客园 (cnblogs.com)

留下评论