def __repr__(self):
if isinstance(self._parent, MultiplyDistribution):
text = self._parent.__repr__()
else:
text = "{} ({}): {}".format(self._parent.prob_text, self._parent.distribution_name, self._parent.__repr__())
text += "\n"
if isinstance(self._child, MultiplyDistribution):
text += self._child.__repr__()
else:
After Change
" so you must set sum_dim=True.".format(self._parent.prob_text, self._child.prob_text))
def __repr__(self):
return self._parent.__repr__() + "\n" + self._child.__repr__()
class ReplaceVarDistribution(Distribution):