@parse_args

@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这个函数。

猜你喜欢

转载自blog.csdn.net/weixin_44594953/article/details/127972705