def METHOD_NAME(
    self: Float[LinearOperator, "... M N"], batch_shape: Union[torch.Size, List[int]]
) -> Float[LinearOperator, "... M N"]:
    batch_shape = torch.Size((*batch_shape, self.base_linear_op.size(-3)))
    res = self.__class__(self.base_linear_op.METHOD_NAME(batch_shape))
    return res