fix mul_dim

Author: GiggleLiu

Project.toml

@@ -1,7 +1,7 @@
 name = "SimpleTensorNetworks"
 uuid = "4456351a-5be3-4067-ade9-541926a41e04"
 authors = ["GiggleLiu <[email protected]> and contributors"]
-version = "0.1.1"
+version = "0.1.2"
 
 [deps]
 LightGraphs = "093fc24a-ae57-5d10-9952-331d41423f4d"

src/tensors.jl

@@ -43,7 +43,7 @@ function mul_dim(t::LabeledTensor, m::AbstractMatrix; dim::Int)
     iA = ntuple(i->i, ndims(data))
     iB = (dim, -dim)
     iC = ntuple(i->i==dim ? -dim : i, ndims(data))
-    LabeledTensor(tensorcontract(iA, data, iB, m, iC; compress=false), t.labels)
+    LabeledTensor(tensorcontract(iA, data, iB, m, iC), t.labels)
 end
 
 struct PlotMeta

test/tensorcontract.jl

@@ -40,3 +40,15 @@ end
     @test tnet.tensors[] ≈ tOut2
     @test contracted_labels == [1, 2]
 end
+
+@testset "mul_dim" begin
+    A = zeros(Float64, 10, 32, 21);
+    B = zeros(Float64, 32, 11);
+    tA = LabeledTensor(A, [1,2,3])
+    tB = LabeledTensor(B, [2,4])
+    tA1 = SimpleTensorNetworks.mul_dim(tA, B; dim=2)
+    tA2 = tA * tB
+    @test tA1.array ≈ permutedims(tA2.array, (1,3,2))
+    @test tA1.labels == [1, 2, 3]
+    @test tA2.labels == [1, 3, 4]
+end