Commit 8613ae95 authored by Martin Reinecke's avatar Martin Reinecke
Browse files

fix

parent 7da95882
Pipeline #14310 passed with stage
in 5 minutes and 20 seconds
......@@ -104,7 +104,7 @@ class RGSpace(Space):
preserve_gaussian_variance=False):
# check axes
if axes is None:
axes = range(len(x.shape))
axes = range(len(self.shape))
assert len(x.shape) >= len(self.shape), "shapes mismatch"
assert len(axes) == len(self.shape), "axes mismatch"
......
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