def sample(self, num_sample): """ draws a sample from classes based on weights """ return t.multinomial(self.weights, num_sample, True)