Add (alias, zeros) function

ITensor · Aug 1, 2023 · 861cedf · 861cedf
1 parent 412c878
commit 861cedf
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/NDTensors/src/zeros/zeros.jl b/NDTensors/src/zeros/zeros.jl
@@ -36,3 +36,11 @@ setindex!(A::NDTensors.Zeros, v, I) = setindex!(A.z, v, I)
 
 Base.iszero(t::Tensor) = iszero(storage(t))
 Base.iszero(st::TensorStorage) = data(st) isa Zeros
+
+function (arraytype::Type{<:Zeros})(::AllowAlias, A::Zeros)
+  return A
+end
+
+function (arraytype::Type{<:Zeros})(::NeverAlias, A::Zeros)
+  return copy(A)
+end