Есть ли способ обновить разреженные частичные индексы на лету?

Допустим, у меня есть этот явный компонент с разреженным частичным объявлением. Есть ли способ изменить частичные индексы после того, как я настрою проблему и начну оптимизацию. Размер партиалов не изменится (скажем, всегда (4,1)), но расположение (строки/столбцы) изменится. Возможно ли это, может быть, есть хитрость?

import numpy as np
import openmdao.api as om
class SparsePartialComp(om.ExplicitComponent):
    def setup(self):
        self.add_input('x', shape=(4,))
        self.add_output('f', shape=(2,))

        self.declare_partials(of='f', wrt='x',
                              rows=[0, 1, 1, 1],
                              cols=[0, 1, 2, 3])

    def compute_partials(self, inputs, partials):
        # Corresponds to the [(0,0), (1,1), (1,2), (1,3)] entries.
        partials['f', 'x'] = [1., 2., 3., 4.]

model = om.Group()
comp = om.IndepVarComp()
comp.add_output('x', np.ones(4))

model.add_subsystem('input', comp)
model.add_subsystem('example', SparsePartialComp())

model.connect('input.x', 'example.x')

problem = om.Problem(model=model)
problem.setup()
problem.run_model()
totals = problem.compute_totals(['example.f'], ['input.x'])

print(totals['example.f', 'input.x'])