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
Pipeline #10026 passed with stages
in 17 minutes and 43 seconds
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment