diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp
index 005175ab613594c48959073c4674e6d69b60b29f..143c1d6e13afc08fddd5d6a6ab57792240433784 100644
--- a/python_binding/data/pybind_Tensor.cpp
+++ b/python_binding/data/pybind_Tensor.cpp
@@ -85,6 +85,7 @@ void init_Tensor(py::module& m){
     .def("dtype", &Tensor::dataType)
     .def("init_grad", &Tensor::initGrad)
     .def("size", &Tensor::size)
+    .def("capacity", &Tensor::capacity)
     .def("resize", (void (Tensor::*)(const std::vector<DimSize_t>&, std::vector<DimSize_t>)) &Tensor::resize)
     .def("has_impl", &Tensor::hasImpl)
     .def("get_coord", &Tensor::getCoord)