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