def _init_weight(self): init.xavier_uniform(self.transitions) self.transitions.data[START, :].fill_(-10000.) self.transitions.data[:, STOP].fill_(-10000.)