看torchvison源码时,发现如下注释宏:
"# type"为python注释宏,添加该注释宏后,解释器会把被注释变量解释成对应类型,而不管变量本身类型。
单值注释:
a = 1 #type:list
a.append() #这里我们.的时候会自动跳出list的内置方法
多值注释:
x,y = 1,2 #type:str 两个都是字符串
x,y = 1,2 #type:str,int 两个不同类型
函数注释:
def forward(self, x):
# type: (int) -> None 输入参数为int,函数返回None类型
return None
很直观,torchvison源码中出现的为函数注释。输入为Tensor,输出GoogLeNetOutputs类型。