def METHOD_NAME(
    self: Float[LinearOperator, "... M N"], batch_shape: Union[torch.Size, List[int]]
) -> Float[LinearOperator, "... M N"]:
    if len(batch_shape) == 0:
        return self
    return self.__class__(self.root.METHOD_NAME(batch_shape))