Add pipe symbol

software-mansion-labs · Jul 4, 2024 · cc898ad · cc898ad
1 parent 81346e3
commit cc898ad
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()