def trace_with_covar(self, A, component_index, latent_index): assert component_index == 0 return util.tr_AB(A, self.covars[latent_index])
def tr_AS(self, A, k, j): return tr_AB(A, self.s[k, j, :])
def tr_AinvS(self, L, k, j): a = solve_triangular(L, self.L[k, j, :], lower=True) return tr_AB(a.T, a)