Skip to content

Commit 0aeca66

Browse files
committed
show size
1 parent da61225 commit 0aeca66

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/fusiontensor/base_interface.jl

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,10 +122,13 @@ function Base.similar(::FusionTensor, ::Type{T}, new_axes::FusionTensorAxes) whe
122122
return FusionTensor{T}(undef, new_axes)
123123
end
124124

125-
Base.show(io::IO, ft::FusionTensor) = print(io, "$(ndims(ft))-dim FusionTensor")
125+
function Base.show(io::IO, ft::FusionTensor)
126+
return print(io, "$(ndims(ft))-dim FusionTensor with size $(size(ft))")
127+
end
126128

127129
function Base.show(io::IO, ::MIME"text/plain", ft::FusionTensor)
128-
print(io, "$(ndims(ft))-dim FusionTensor with axes:")
130+
print(io, ft)
131+
print(" and axes:")
129132
for ax in axes(ft)
130133
print(io, "\n", ax)
131134
end

0 commit comments

Comments
 (0)