def METHOD_NAME(self, expand=True):
    num_events = self._num_events
    values = torch.arange(num_events, dtype=torch.long, device=self._param.device)
    values = values.view((-1,) + (1,) * len(self._batch_shape))
    if expand:
        values = values.expand((-1,) + self._batch_shape)
    return values