def __init__(self, d, k): self.m = pf.Parameter([d, k]) self.s = pf.ScaleParameter([d, k]) self.w = pf.DirichletParameter(k)
def __init__(self): self.probs = pf.DirichletParameter(k=3, name='probs')