Explicitly set step in range

pnezis · Dec 11, 2024 · e7f9ce2 · e7f9ce2
1 parent a004499
commit e7f9ce2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/tucan/image.ex b/lib/tucan/image.ex
@@ -45,8 +45,10 @@ defmodule Tucan.Image do
         :lower -> {height - 1, 0}
       end
 
+    step = if start_index <= end_index, do: 1, else: -1
+
     y =
-      Nx.tensor(Enum.to_list(start_index..end_index))
+      Nx.tensor(Enum.to_list(start_index..end_index//step))
       |> Nx.broadcast({height, width}, axes: [0])
       |> Nx.to_flat_list()