Commit 70e096aa authored by Jait Dixit's avatar Jait Dixit
Browse files

Correct reshaping in weight method

parent 5405547d
Pipeline #10022 passed with stage
in 14 minutes and 38 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