define norm p

Author: ogauthe

src/fusiontensor/linear_algebra_interface.jl

@@ -33,13 +33,13 @@ function LinearAlgebra.mul!(
   return C
 end
 
-function LinearAlgebra.norm(ft::FusionTensor)
+function LinearAlgebra.norm(ft::FusionTensor, p::Real=2)
   m = data_matrix(ft)
   row_sectors = sectors(codomain_axis(ft))
   n2 = sum(eachblockstoredindex(m); init=zero(real(eltype(ft)))) do b
-    return quantum_dimension(row_sectors[Int(first(Tuple(b)))]) * norm(m[b])^2
+    return quantum_dimension(row_sectors[Int(first(Tuple(b)))]) * norm(m[b], p)^p
   end
-  return sqrt(n2)
+  return n2^(1 / p)
 end
 
 LinearAlgebra.normalize(ft::FusionTensor, p::Real=2) = ft / norm(ft, p)

test/test_linear_algebra.jl

@@ -26,6 +26,9 @@ include("setup.jl")
     ft = to_fusiontensor(sdst, (g, g), (dual(g), dual(g)))
     @test isnothing(check_sanity(ft))
     @test norm(ft) ≈ √3 / 2
+    @test norm(ft, 2) ≈ √3 / 2
+    @test norm(ft, 2.0) ≈ √3 / 2
+    @test norm(ft, 1) ≈ 1.5
     @test isapprox(tr(ft), 0; atol=eps(Float64))
 
     ft2 = normalize(ft)