improving tensor print formating

Author: fschlimb

src/DDPTensorImpl.cpp

@@ -95,13 +95,13 @@ std::string DDPTensorImpl::__repr__() const
     const auto nd = ndims();
     std::ostringstream oss;
     oss << "ddptensor{gs=(";
-    for(auto i=0; i<nd; ++i) oss << _gs_aligned[i] << ", ";
+    for(auto i=0; i<nd; ++i) oss << _gs_aligned[i] << (i==nd-1 ? "" : ", ");
     oss << "), loff=(";
-    for(auto i=0; i<nd; ++i) oss << _lo_aligned[i] << ", ";
+    for(auto i=0; i<nd; ++i) oss << _lo_aligned[i] << (i==nd-1 ? "" : ", ");
     oss << "), lsz=(";
-    for(auto i=0; i<nd; ++i) oss << _sizes[i] << ", ";
+    for(auto i=0; i<nd; ++i) oss << _sizes[i] << (i==nd-1 ? "" : ", ");
     oss << "), str=(";
-    for(auto i=0; i<nd; ++i) oss << _strides[i] << ", ";
+    for(auto i=0; i<nd; ++i) oss << _strides[i] << (i==nd-1 ? "" : ", ");
     oss << "), p=" << _allocated << ", poff=" << _offset << "}\n";
 
     dispatch(_dtype, _aligned, [this, nd, &oss](auto * ptr) {

src/include/ddptensor/DDPTensorImpl.hpp

@@ -153,14 +153,13 @@ class DDPTensorImpl : public tensor_i
         if(d==ndims()-1) {
             oss << "[";
             for(auto i=0; i<sz; ++i) {
-                oss << cptr[i*stride];
-                if(i<sz-1) oss << " ";
+                oss << cptr[i*stride] << (i<sz-1 ? " " : "");
             }
             oss << "]";
         } else {
             oss << "[";
             for(auto i=0; i<sz; ++i) {
-                if(i) for(auto x=0; x<=d; ++x) oss << " ";
+                if(i) oss << std::string(d+1, ' ');
                 printit(oss, d+1, cptr);
                 if(i<sz-1) oss << "\n";
                 cptr += stride;