fixed bug in understanding of gather

cwright7101 · Sep 30, 2019 · 8328985 · 8328985
1 parent d01d919
commit 8328985
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/zonoml/zonoml_fun.c b/zonoml/zonoml_fun.c
@@ -427,8 +427,9 @@ elina_abstract0_t *handle_gather_layer(elina_manager_t* man, bool destructive, e
 	elina_dimperm_t * dimperm = elina_dimperm_alloc(size);
 	size_t i;
 	for(i=0; i < size; i++){
-          dimperm->dim[i] = (elina_dim_t)indexes[i];
-        }
+
+		dimperm->dim[indexes[i]] = i;
+	}
 	elina_abstract0_t *res = elina_abstract0_permute_dimensions(man,destructive, abs,dimperm);
 	elina_dimperm_free(dimperm);
 	return res;