def METHOD_NAME(self: Float[LinearOperator, "... M N"]) -> Float[torch.Tensor, "... N"]:
    block_diag = self.base_linear_op.METHOD_NAME()
    return block_diag.mT.contiguous().view(*block_diag.shape[:-2], -1)