Parameter(torch.randn(len(self.op_choices)) * 1e-3) def forward(self, *args, ... return torch.argsort(-self.alpha).cpu().numpy().tolist()[:self.n_chosen].
確定! 回上一頁