def ell(self, mu, sigma, Y): return cross_ent_normal(mu, np.diag(sigma), Y, self.sigma)
def ell(self, mu, sigma, Y): return cross_ent_normal(mu, np.diag(sigma), Y, np.array([[self.sigma]]))