def METHOD_NAME(self, batch_shape, _instance=None):
    new = self._get_checked_instance(Geometric, _instance)
    batch_shape = torch.Size(batch_shape)
    if "probs" in self.__dict__:
        new.probs = self.probs.METHOD_NAME(batch_shape)
    if "logits" in self.__dict__:
        new.logits = self.logits.METHOD_NAME(batch_shape)
    super(Geometric, new).__init__(batch_shape, validate_args=False)
    new._validate_args = self._validate_args
    return new