From ab12e41578297630932e7f8604f8dc1476f3b838 Mon Sep 17 00:00:00 2001
From: Liao Junxuan <mikeljx@126.com>
Date: Sun, 23 Oct 2022 19:07:28 +0800
Subject: [PATCH] fix: don't call batch() if batch == False

---
 tenseal/cpp/tensors/ckkstensor.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tenseal/cpp/tensors/ckkstensor.cpp b/tenseal/cpp/tensors/ckkstensor.cpp
index 8390822e..1bae66fd 100644
--- a/tenseal/cpp/tensors/ckkstensor.cpp
+++ b/tenseal/cpp/tensors/ckkstensor.cpp
@@ -17,9 +17,10 @@ CKKSTensor::CKKSTensor(const shared_ptr<TenSEALContext>& ctx,
 
     vector<Ciphertext> enc_data;
     vector<size_t> enc_shape = tensor.shape();
-    auto data = tensor.batch(0);
+    decltype(tensor.batch(0)) data;
     size_t size;
     if (batch) {
+        data = tensor.batch(0);
         _batch_size = enc_shape[0];
         enc_shape.erase(enc_shape.begin());
         size = tensor.batch(0).size();