def generate_sample(self, base_target, std_val=1.0):
        """Generate a random sample around the given point.

        The standard deviation is in pixel.
        """
        noise = std_val * torch.rand_like(base_target)
        return base_target + noise
Generate a random sample around the given point.

        The standard deviation is in pixel.
        
tests/integration/test_soft_argmax2d.py