Skip to content

Commit 1dcab3c

Browse files
committed
specialize on BlockedTrivialPermutation
1 parent fe9f9cd commit 1dcab3c

File tree

1 file changed

+3
-11
lines changed

1 file changed

+3
-11
lines changed

src/fusiontensor/tensor_algebra_interface.jl

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -41,20 +41,12 @@ struct FusionTensorFusionStyle <: FusionStyle end
4141

4242
TensorAlgebra.FusionStyle(::Type{<:FusionTensor}) = FusionTensorFusionStyle()
4343

44-
function TensorAlgebra.matricize(
45-
::FusionTensorFusionStyle, ft::AbstractArray, biperm::AbstractBlockPermutation{2}
46-
)
47-
permuted = permutedims(ft, biperm)
48-
return FusionTensor(
49-
data_matrix(permuted), (codomain_axis(permuted),), (domain_axis(permuted),)
50-
)
51-
end
52-
53-
# lift ambiguity
5444
function TensorAlgebra.matricize(
5545
::FusionTensorFusionStyle, ft::AbstractArray, biperm::BlockedTrivialPermutation{2}
5646
)
57-
return matricize(FusionTensorFusionStyle(), ft, blockedperm(BlockedTuple(tbp)))
47+
blocklengths(biperm) == blocklengths(axes(ft)) ||
48+
throw(ArgumentError("Invalid trivial biperm"))
49+
return FusionTensor(data_matrix(ft), (codomain_axis(ft),), (domain_axis(ft),))
5850
end
5951

6052
function TensorAlgebra.unmatricize(::FusionTensorFusionStyle, m, blocked_axes)

0 commit comments

Comments
 (0)