예제 #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)