def _compute_weights_to_parent(self, index, weights): return misc.put_simple( weights, self._indices, axis=self._plate_axis, length=self._original_length, )
def _compute_message_to_parent(self, index, m_child, u_parent): m = [ misc.put_simple( mi, self._indices, axis=self._plate_axis - len(dimi), length=self._original_length, ) for (mi, dimi) in zip(m_child, self.dims) ] return m
def _compute_message_to_parent(self, index, m_child, u_parent): m = [ misc.put_simple( mi, self._indices, axis=self._plate_axis-len(dimi), length=self._original_length, ) for (mi, dimi) in zip(m_child, self.dims) ] return m