matrix-free cg中如何传入更多变量?

文档中写的,matrixfree cg,需要提供一个compute_Ax的kernel函数,但是这个函数只有两个输入参数,也就是v和mv。可是我的程序计算Ax可能需要更多的参数,例如我的函数需要用的class里面的参数,所以我的函数可能是
def compute_Ax(self, v:ti.template(), mv:ti.template()):
这种函数应该是不能传入到LinearOperator吧?但是对于很多模拟问题也很难把所有计算Ax的东西放到一个函数里写,这种情况要怎么处理呢?