From 430fb66a92003e13fa187887948224e8f9100768 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Thu, 13 Jun 2024 07:38:58 +0000
Subject: [PATCH] Bind Tensor::capacity() method.
---
python_binding/data/pybind_Tensor.cpp | 1 +
1 file changed, 1 insertion(+)
diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp
index 005175ab6..143c1d6e1 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)
--
GitLab