Method cuda

pycontrast/memory/alias_multinomial.py:45–47

(self)

            self.prob[last_one] = 1

    def cuda(self):
        self.prob = self.prob.cuda()
        self.alias = self.alias.cuda()

    def draw(self, N):
        """