def _initialize(self, arr): tmp = kaiming_normal_(arr, self.a, self.mode, self.nonlinearity) _assignment(arr, tmp)
def _initialize(self, arr): init._assignment(arr, 2)
def _initialize(self, arr): tmp = xavier_uniform_(arr, self.gain) _assignment(arr, tmp)