Method sample

imperative/python/megengine/module/vision.py:567–568

(self, batch)

        self.beta = beta

    def sample(self, batch):
        return self.beta_func(self.beta, self.beta, size=(batch,))

    def forward(self, data1, label1, data2, label2):
        assert all(