[param concatenation] allows assignmend more robustly

This commit is contained in:
Max Zwiessele 2014-05-09 14:07:34 +01:00
parent 457e00f058
commit 4eac8a5965

View file

@ -316,8 +316,8 @@ class ParamConcatenation(object):
val = val.values()
ind = numpy.zeros(sum(self._param_sizes), dtype=bool); ind[s] = True;
vals = self.values(); vals[s] = val
[numpy.copyto(p, vals[ps], where=ind[ps])
for p, ps in zip(self.params, self._param_slices_)]
for p, ps in zip(self.params, self._param_slices_):
p.flat[ind[ps]] = vals[ps]
if update:
self.update_all_params()
def values(self):