diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp index 973fc6f9a94d108d8b81c93384ef8468d8247c41..2171d48975db8f4029abe7982bf6dfc17640dd52 100644 --- a/python_binding/data/pybind_Tensor.cpp +++ b/python_binding/data/pybind_Tensor.cpp @@ -345,7 +345,7 @@ void init_Tensor(py::module& m){ return b.toString(); }) .def("__repr__", [](Tensor& b) { - return fmt::format("Tensor(dims = {}, dtype = {})", b.dims(), std::string(EnumStrings<DataType>::data[static_cast<int>(b.dataType())])); + return fmt::format("Tensor({}, dims = {}, dtype = {})", b.toString(-1, 7), b.dims(), b.dataType()); }) .def("__len__", [](Tensor& b) -> size_t{ return b.size();