Add pipe symbol

software-mansion-labs · Jul 3, 2024 · 5901895 · 5901895
1 parent 418def9
commit 5901895
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lib/ex_vision/utils.ex b/lib/ex_vision/utils.ex
@@ -86,8 +86,7 @@ defmodule ExVision.Utils do
 
   defp ensure_grad_3(tensor) do
     tensor
-
-    Nx.rank()
+    |> Nx.rank()
     |> case do
       3 -> [tensor]
       4 -> tensor |> Nx.to_batched(1) |> Stream.map(&Nx.squeeze(&1, axes: [0])) |> Enum.to_list()