[NDTensors] More compact tensor printing

NDTensors/Project.toml

@@ -1,7 +1,7 @@
 name = "NDTensors"
 uuid = "23ae76d9-e61a-49c4-8f12-3f1a16adf9cf"
 authors = ["Matthew Fishman <mfishman@flatironinstitute.org>"]
-version = "0.4.15"
+version = "0.4.16"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"

NDTensors/src/dense/densetensor.jl

@@ -319,5 +319,7 @@ end
 
 function Base.show(io::IO, mime::MIME"text/plain", T::DenseTensor)
     summary(io, T)
-    return print_tensor(io, T)
+    print_tensor(io, T)
+    println(io)
+    return nothing
 end

NDTensors/src/diag/diagtensor.jl

@@ -217,5 +217,6 @@ end
 function Base.show(io::IO, mime::MIME"text/plain", T::DiagTensor)
     summary(io, T)
     print_tensor(io, T)
+    println(io)
     return nothing
 end

NDTensors/src/tensor/tensor.jl

@@ -495,5 +495,9 @@ end
 # Printing
 #
 
-print_tensor(io::IO, T::Tensor) = Base.print_array(io, expose(T))
-print_tensor(io::IO, T::Tensor{<:Number, 1}) = Base.print_array(io, reshape(T, (dim(T), 1)))
+function print_tensor(io::IO, T::Tensor)
+    return Base.print_array(IOContext(io, :limit => true), expose(T))
+end
+function print_tensor(io::IO, T::Tensor{<:Number, 1})
+    return Base.print_array(IOContext(io, :limit => true), reshape(T, (dim(T), 1)))
+end

src/itensor.jl

@@ -2025,8 +2025,16 @@ end
 # TODO: make a specialized printing from Diag
 # that emphasizes the missing elements
 function show(io::IO, T::ITensor)
-    println(io, "ITensor ord=$(order(T))")
-    return show(io, MIME"text/plain"(), tensor(T))
+    if get(io, :compact, false)
+        # Just show the indices in compact view, used in some
+        # cases when printing arrays of ITensors (similar to
+        # printing of MPS in ITensorMPS.jl).
+        show(io, inds(T))
+    else
+        println(io, "ITensor ord=$(order(T))")
+        show(io, MIME"text/plain"(), tensor(T))
+    end
+    return nothing
 end
 
 function show(io::IO, mime::MIME"text/plain", T::ITensor)