From 8f2e2711ea50c3e13c20a5679ceb0b7fcce06ed4 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Sat, 27 Nov 2021 11:58:09 +0800
Subject: [PATCH] fix(pylite): add dtype to empty tensor to_numpy api

GitOrigin-RevId: 4f61e5b544e1a52ed7ddf9abf15c042ac3465bb6
---
 lite/pylite/megenginelite/tensor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lite/pylite/megenginelite/tensor.py b/lite/pylite/megenginelite/tensor.py
index 0af8f27f..ef86154f 100644
--- a/lite/pylite/megenginelite/tensor.py
+++ b/lite/pylite/megenginelite/tensor.py
@@ -421,7 +421,8 @@ class LiteTensor(object):
         """
         self.update()
         if self.nbytes <= 0:
-            return np.array([])
+            np_type = _lite_type_to_nptypes[LiteDataType(self._layout.data_type)]
+            return np.array([], dtype=np_type)
         if self.is_continue and (
             self.is_pinned_host or self.device_type == LiteDeviceType.LITE_CPU
         ):