Skip to content
Snippets Groups Projects

[UPD] Tensor formatting

Merged Maxence Naud requested to merge feat_precision-log into dev
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -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();
Loading