define zeros and randn

Author: ogauthe

Project.toml

@@ -1,7 +1,7 @@
 name = "FusionTensors"
 uuid = "e16ca583-1f51-4df0-8e12-57d32947d33e"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.4.1"
+version = "0.4.2"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
@@ -11,6 +11,7 @@ GradedArrays = "bc96ca6e-b7c8-4bb6-888e-c93f838762c2"
 HalfIntegers = "f0d1745a-41c9-11e9-1dd9-e5d34d218721"
 LRUCache = "8ac3fa9e-de4c-5943-b1dc-09c6b5f20637"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Strided = "5e0ebb24-38b0-5f93-81fe-25c709ecae67"
 TensorAlgebra = "68bd88dc-f39d-4e12-b2ca-f046b68fcc6a"
 TensorProducts = "decf83d6-1968-43f4-96dc-fdb3fe15fc6d"
@@ -25,6 +26,7 @@ GradedArrays = "0.4.13"
 HalfIntegers = "1.6"
 LRUCache = "1.6"
 LinearAlgebra = "1.10"
+Random = "1.11.0"
 Strided = "2.3"
 TensorAlgebra = "0.3.8"
 TensorProducts = "0.1.7"

src/fusiontensor/base_interface.jl

@@ -1,8 +1,8 @@
 # This files defines Base functions for FusionTensor
 
 using Accessors: @set
-
-using BlockSparseArrays: @view!
+using Random: randn!
+using BlockSparseArrays: @view!, eachstoredblock
 using TensorAlgebra: BlockedTuple, tuplemortar
 
 set_data_matrix(ft::FusionTensor, data_matrix) = @set ft.data_matrix = data_matrix
@@ -81,6 +81,15 @@ end
 
 Base.permutedims(ft::FusionTensor, args...) = fusiontensor_permutedims(ft, args...)
 
+function Base.randn(::Type{T}, fta::FusionTensorAxes) where {T}
+  ft = FusionTensor(T, fta)
+  for m in eachstoredblock(data_matrix(ft))
+    m = randn!(m)
+  end
+  return ft
+end
+Base.randn(fta::FusionTensorAxes) = randn(Float64, fta)
+
 function Base.similar(ft::FusionTensor, T::Type)
   # reuse trees_block_mapping
 
@@ -124,3 +133,6 @@ function Base.view(ft::FusionTensor, f1::SectorFusionTree, f2::SectorFusionTree)
   charge_matrix = @view! data_matrix(ft)[trees_block_mapping(ft)[f1, f2]]
   return reshape(charge_matrix, charge_block_size(ft, f1, f2))
 end
+
+Base.zeros(::Type{T}, fta::FusionTensorAxes) where {T} = FusionTensor(T, fta)
+Base.zeros(fta::FusionTensorAxes) = zeros(Float64, fta)

test/test_basics.jl

@@ -35,6 +35,12 @@ include("setup.jl")
   g1 = gradedrange([U1(0) => 1, U1(1) => 2, U1(2) => 3])
   g2 = dual(gradedrange([U1(0) => 2, U1(1) => 2, U1(3) => 1]))
 
+  fta = FusionTensorAxes((g1,), (g2,))
+  ft0 = FusionTensor(Float64, fta)
+  @test ft0 isa FusionTensor
+  @test space_isequal(codomain_axis(ft0), g1)
+  @test space_isequal(domain_axis(ft0), g2)
+
   # check dual convention when initializing data_matrix
   ft0 = FusionTensor(Float64, (g1,), (g2,))
   @test ft0 isa FusionTensor
@@ -264,6 +270,13 @@ end
   @test_throws DimensionMismatch ft7 + ft3
   @test_throws DimensionMismatch ft7 - ft3
   @test_throws DimensionMismatch ft7 * ft3
+
+  fta = FusionTensorAxes((g1,), (g2, g2))
+  @test zeros(fta) isa FusionTensor{Float64,3}
+  @test zeros(ComplexF64, fta) isa FusionTensor{ComplexF64,3}
+  ft9 = randn(ComplexF64, fta)
+  @test ft9 isa FusionTensor{ComplexF64,3}
+  @test all(data_matrix(ft9)[Block(1, 6)] .!= 0)
 end
 
 @testset "missing SectorProduct" begin