示例#1
0
文件: auto.py 项目: yuntang/pytorch
 def __init__(self, *args, **kwargs):
     Function.__init__(self)
     self.weight = kwargs.get('weight')
     self.additional_args = list(args)
示例#2
0
文件: auto.py 项目: yuntang/pytorch
 def __init__(self, *args):
     if is_inplace:
         InplaceFunction.__init__(self, args[-1])
     else:
         Function.__init__(self)
     self.additional_args = list(args)