def trace_with_covar(self, A, component_index, latent_index):
     assert component_index == 0
     return util.tr_AB(A, self.covars[latent_index])
Ejemplo n.º 2
0
 def tr_AS(self, A, k, j):
     return tr_AB(A, self.s[k, j, :])
Ejemplo n.º 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)
Ejemplo n.º 4
0
 def trace_with_covar(self, A, component_index, latent_index):
     assert component_index == 0
     return util.tr_AB(A, self.covars[latent_index])
Ejemplo n.º 5
0
 def tr_AS(self, A, k, j):
     return tr_AB(A, self.s[k, j, :])
Ejemplo n.º 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)