From cc898ade560816c39ad1f446a67cc0d5d40c46ad Mon Sep 17 00:00:00 2001
From: msluszniak <mateusz.sluszniak@swmansion.com>
Date: Wed, 3 Jul 2024 13:46:50 +0000
Subject: [PATCH] Add pipe symbol

---
 lib/ex_vision/utils.ex | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/lib/ex_vision/utils.ex b/lib/ex_vision/utils.ex
index 1003257..e8813c3 100644
--- 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()