Sorry, I was a bit unclear. Is this handled correctly in the accuracy?
So your check is whether there is only one dimension with shape != 1.
Checking just that might lead to surprising results:

```
In [4]: np.mean(np.array([[0, 1, 2]]) == np.array([[0], [1], [2]]))
Out[4]: 0.33333333333333331
```