Skip to content

Commit c06cb6c

Browse files
committed
Add similar(::DiagonalTensorMap, [::Type{T}]) -> DiagonalTensorMap
1 parent b4f4dfc commit c06cb6c

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/tensors/diagonal.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,11 @@ function DiagonalTensorMap(t::AbstractTensorMap{T,S,1,1}) where {T,S}
7676
return d
7777
end
7878

79+
Base.similar(d::DiagonalTensorMap) = DiagonalTensorMap(similar(d.data), d.domain)
80+
function Base.similar(d::DiagonalTensorMap, ::Type{T}) where {T<:Number}
81+
return DiagonalTensorMap(similar(d.data, T), d.domain)
82+
end
83+
7984
# TODO: more constructors needed?
8085

8186
# Special case adjoint:

0 commit comments

Comments
 (0)