矩阵关于矩阵的自动求导

在做隐式有限元时,有个很重要的物理量是Jaccobian矩阵。例如应力张量对应变张量的偏导。若我有一个应变场 strain=ti.Matrix.field(3,3,float,shape=(X,Y,Z)) 和应力场 stress=field(3,3,float,shape=(X,Y,Z)),有一个kernel能够使用每个位置点的strain的3x3矩阵求出该点的stress 3x3矩阵。现在想求\partial stress/\partial strain,每个位置点上一个3x3x3x3的矩阵。想请教两个问题,Taichi目前没办法定义高阶张量,如何定义该Jaccobian矩阵场,使用3x3x3x3xXxYxZ的标量场来计算?若是想使用自动求导方案,不知道有没有Taichi的范式可以参考一下?