经常在pytorch中看到带有“下划线”的方法,个人总结了下,大概是这两类:
1.放在方法名前 2.放在方法名后

  1. 自定义类中,含有的方法名带有下划线【放在方法名前】
    在这里插入图片描述
    作用:会阻止其他python脚本通过【from module import *】语句导入该名字,即该名字不会被星号匹配;保证只在该类中进行调用。

  2. 框架中,带有下划线的方法【放在方法名后】
    在这里插入图片描述
    作用:对Tensor进行in-place=True的修改。
    【in-place=True:在一个tensor上操作后,直接修改这个tensor本身】
    【in-place=False:在一个tensor上操作后,不修改这个tensor本身,生成一个修改后新的Tensor】
    sp:平时我们看到大多都是没有加上“_”的方法,这些都属于in-place=False。

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐