def METHOD_NAME(
    self: Float[LinearOperator, "*batch M N"],
    rhs: Union[Float[Tensor, "*batch2 M P"], Float[LinearOperator, "*batch2 M P"]],
) -> Union[Float[LinearOperator, "... N P"], Float[Tensor, "... N P"]]:
    rhs_expanded = self._expand(rhs, self.row_mask)
    res_expanded = self.base.METHOD_NAME(rhs_expanded)
    res = res_expanded[..., self.col_mask, :]
    return res