diff --git a/x2paddle/__init__.py b/x2paddle/__init__.py
index b6a4e42f7aab145d029dcae03fa06a2ba23a3525..b1d07005b4633b6bd7fec4873a986ee23485ae7d 100644
--- a/x2paddle/__init__.py
+++ b/x2paddle/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "1.0.1"
+__version__ = "1.0.2"
 
 from .core.program import PaddleGraph
 
diff --git a/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py b/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py
index d33e84a8294f99866a0cfac2d2fe863718d50a6c..931ace6efd47ce3e11ae0e2ce9f8ab84046ab989 100644
--- a/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py
+++ b/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py
@@ -281,7 +281,7 @@ class PyTorchOpMapper(OpMapper):
             scope_name=scope_name,
             data=output_name)
         if self.input_examples is not None:
-            input_np = self.input_examples[input_ct].detach().numpy()
+            input_np = self.input_examples[input_ct].cpu().detach().numpy()
             self.inputs_info[
                 output_name] = [list(input_np.shape), str(input_np.dtype)]
         return [], [output_name]