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_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
