fix hist

Signed-off-by: weiwee <wbwmat@gmail.com> Signed-off-by: sagewe <wbwmat@gmail.com>
FederatedAI · Jul 21, 2023 · 9e70e1e · 9e70e1e
1 parent b0d52f0
commit 9e70e1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/fate/arch/tensor/inside/_op_hist.py b/python/fate/arch/tensor/inside/_op_hist.py
@@ -10,7 +10,7 @@ def update(self, features, labels):
         shape_x, shape_y = features.shape
         for i in range(shape_x):
             for j in range(shape_y):
-                v = features[i, j]
+                v = features[i, j].item()
                 if j not in self.data:
                     self.data[j] = {}
                 if v not in self.data[j]: