Skip to content
Snippets Groups Projects
Commit e5d6ef9c authored by Theo Steininger's avatar Theo Steininger
Browse files

Merge branch 'glspace_fixes' into 'master'

Correct reshaping in weight method

See merge request !48
parents fa788a4e 70e096aa
No related branches found
No related tags found
1 merge request!48Correct reshaping in weight method
Pipeline #
......@@ -139,8 +139,8 @@ class GLSpace(Space):
if axes is not None:
# reshape the weight array to match the input shape
new_shape = np.ones(len(x.shape), dtype=np.int)
for index in range(len(axes)):
new_shape[index] = len(weight)
# we know len(axes) is always 1
new_shape[axes[0]] = len(weight)
weight = weight.reshape(new_shape)
if inplace:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment