diff --git a/colossalai/fx/proxy.py b/colossalai/fx/proxy.py
index 3299b14a4..ee2444d0d 100644
--- a/colossalai/fx/proxy.py
+++ b/colossalai/fx/proxy.py
@@ -57,6 +57,10 @@ class ColoProxy(Proxy):
         self._assert_meta_data_is_tensor()
         return self.meta_data.shape
 
+    @property
+    def ndim(self):
+        return self.dim()
+
     def dim(self):
         self._assert_meta_data_is_tensor()
         return self.meta_data.dim()