Return X when src==dst for moveaxis()

vlad-perevezentsev · vlad-perevezentsev · commit 13708353a394 · 2023-08-14T22:33:55.000+02:00
diff --git a/dpctl/tensor/_manipulation_functions.py b/dpctl/tensor/_manipulation_functions.py
@@ -885,6 +885,9 @@ def moveaxis(X, source, destination):
             "the same number of elements"
         )
 
+    if source == destination:
+        return X
+
     ind = [n for n in range(X.ndim) if n not in source]
 
     for src, dst in sorted(zip(destination, source)):

Original file line number	Diff line number	Diff line change
`@@ -885,6 +885,9 @@ def moveaxis(X, source, destination):`
`885`	`885`	`"the same number of elements"`
`886`	`886`	`)`
`887`	`887`
	`888`	`+ if source == destination:`
	`889`	`+ return X`
	`890`	`+`
`888`	`891`	`ind = [n for n in range(X.ndim) if n not in source]`
`889`	`892`
`890`	`893`	`for src, dst in sorted(zip(destination, source)):`