コード例 #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)