Closed seanlatias closed 4 years ago
This solves part of the problem in #155. Several tests are added to check the correctness under arbitrary shapes.
The issue comes from the wrong index flattening given a tuple of indices and the shape. It turns out to be just a single line fix.
This solves part of the problem in #155. Several tests are added to check the correctness under arbitrary shapes.
The issue comes from the wrong index flattening given a tuple of indices and the shape. It turns out to be just a single line fix.