@parse_args用于指定函数中变量的类型,
@parse_args('v') #说明不改变x的类型,如果x是tensor就还是tensor
def custom_sigmoid_onnx(g, x):
@parse_args('i') #说明x是int类型的数据
def custom_sigmoid_onnx(g, x):
@parse_args('i',‘f’) #说明x是int类型的数据,y是float类型的数据
def custom_sigmoid_onnx(g, x,y):
如果查看源码的话,最主要的是_parse_arg这个函数。