define matrix functions

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.5.9"
+version = "0.5.11"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
@@ -28,7 +28,7 @@ LRUCache = "1.6"
 LinearAlgebra = "1.10"
 Random = "1.10"
 Strided = "2.3"
-TensorAlgebra = "0.4"
+TensorAlgebra = "0.4.1"
 TensorProducts = "0.1.7"
 TypeParameterAccessors = "0.4"
 WignerSymbols = "2.0.0"

src/fusiontensor/tensor_algebra_interface.jl

@@ -14,6 +14,7 @@ using TensorAlgebra:
   Matricize,
   blockedperm,
   genperm,
+  matricize,
   unmatricize
 
 function TensorAlgebra.output_axes(
@@ -70,3 +71,17 @@ function TensorAlgebra.unmatricizeadd!(a_dest::FusionTensor, a_dest_mat, invbipe
   data_matrix(a_dest) .= α .* data_matrix(a12) .+ β .* data_matrix(a_dest)
   return a_dest
 end
+
+for f in TensorAlgebra.MATRIX_FUNCTIONS
+  @eval begin
+    function TensorAlgebra.$f(
+      a::FusionTensor, biperm::AbstractBlockPermutation{2}; kwargs...
+    )
+      a_mat = matricize(a, biperm)
+      permuted_axes = axes(a)[biperm]
+      checkspaces_dual(codomain(permuted_axes), domain(permuted_axes))
+      fa_mat = set_data_matrix(a_mat, Base.$f(data_matrix(a_mat); kwargs...))
+      return unmatricize(fa_mat, permuted_axes)
+    end
+  end
+end

test/test_contraction.jl

@@ -1,11 +1,12 @@
 using LinearAlgebra: mul!
-using Test: @test, @testset, @test_broken
+using Test: @test, @testset, @test_broken, @test_throws
 
 using BlockSparseArrays: BlockSparseArray
 using FusionTensors:
-  FusionMatrix, FusionTensor, FusionTensorAxes, domain_axes, codomain_axes
-using GradedArrays: U1, dual, gradedrange
-using TensorAlgebra: contract, matricize, permmortar, tuplemortar, unmatricize, unmatricize!
+  FusionMatrix, FusionTensor, FusionTensorAxes, domain_axes, codomain_axes, to_fusiontensor
+using GradedArrays: SU2, U1, dual, gradedrange
+using TensorAlgebra:
+  TensorAlgebra, contract, matricize, permmortar, tuplemortar, unmatricize, unmatricize!
 
 include("setup.jl")
 
@@ -33,6 +34,54 @@ include("setup.jl")
   @test ft1 ≈ ft2
 end
 
+@testset "Matrix functions" begin
+  sds22 = [
+    0.25 0.0 0.0 0.0
+    0.0 -0.25 0.5 0.0
+    0.0 0.5 -0.25 0.0
+    0.0 0.0 0.0 0.25
+  ]
+  t = reshape(sds22, (2, 2, 2, 2))
+  g2 = gradedrange([SU2(1//2) => 1])
+  ft = to_fusiontensor(t, (g2, g2), (dual(g2), dual(g2)))
+  for f in [
+    :exp,
+    :cis,
+    :log,
+    :sqrt,
+    :cbrt,
+    :cos,
+    :sin,
+    :tan,
+    :csc,
+    :sec,
+    :cot,
+    :cosh,
+    :sinh,
+    :tanh,
+    :csch,
+    :sech,
+    :coth,
+    :acos,
+    :asin,
+    :atan,
+    :acsc,
+    :asec,
+    :acot,
+    :acosh,
+    :asinh,
+    :atanh,
+    :acsch,
+    :asech,
+  ]
+    t2 = @eval reshape(Base.$f(sds22), (2, 2, 2, 2))
+    ft2 = to_fusiontensor(t2, (g2, g2), (dual(g2), dual(g2)))
+    @test (@eval TensorAlgebra.$f(ft, (1, 2, 3, 4), (1, 2), (3, 4))) ≈ ft2
+  end
+  @test_throws ArgumentError TensorAlgebra.exp(ft, (1, 2, 3, 4), (1, 2, 3), (4,))
+  @test_throws ArgumentError TensorAlgebra.exp(ft, (1, 2, 3, 4), (1, 3), (2, 4))
+end
+
 @testset "contraction" begin
   g1 = gradedrange([U1(0) => 1, U1(1) => 2, U1(2) => 3])
   g2 = gradedrange([U1(0) => 2, U1(1) => 2, U1(3) => 1])