Method sample

imperative/python/megengine/module/vision.py:637–638

(self, batch)

        self.custom_func = custom.cutmix_forward()

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

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