diff --git a/iotdb-core/ainode/ainode/core/ingress/iotdb.py b/iotdb-core/ainode/ainode/core/ingress/iotdb.py
index 528c3cb73979..2d1bb74016f4 100644
--- a/iotdb-core/ainode/ainode/core/ingress/iotdb.py
+++ b/iotdb-core/ainode/ainode/core/ingress/iotdb.py
@@ -223,6 +223,7 @@ def __getitem__(self, index):
             + self.seq_len
             + self.output_token_len
         ]
+        result = torch.tensor(result)
         return (
             result[0 : self.seq_len],
             result[self.input_token_len : self.seq_len + self.output_token_len],