Fix balance_matrix3_cpu docstring.