BUG: Fix check for singleton dimensions in einsum.

rsokl · charris · commit 0ba249a554db · 2018-06-01T18:52:52.000-06:00
Was checking array elements rather than array shapes for singleton
dimensions.
diff --git a/numpy/core/einsumfunc.py b/numpy/core/einsumfunc.py
@@ -1109,7 +1109,7 @@ def einsum(*operands, **kwargs):
             # Checks have already been handled
             input_str, results_index = einsum_str.split('->')
             input_left, input_right = input_str.split(',')
-            if 1 in tmp_operands[0] or 1 in tmp_operands[1]:
+            if 1 in tmp_operands[0].shape or 1 in tmp_operands[1].shape:
                 left_dims = {dim: size for dim, size in
                              zip(input_left, tmp_operands[0].shape)}
                 right_dims = {dim: size for dim, size in
