Skip to content

Commit 5a9f043

Browse files
committed
use foreach
1 parent 53a5bb6 commit 5a9f043

File tree

2 files changed

+5
-6
lines changed

2 files changed

+5
-6
lines changed

src/fusiontensor/fusiontensor.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -225,14 +225,14 @@ end
225225
# specific constructors
226226
function Base.zeros(::Type{T}, fta::FusionTensorAxes) where {T}
227227
ft = FusionTensor{T}(undef, fta)
228-
map(m -> fill!(m, zero(T)), eachstoredblock(data_matrix(ft)))
228+
foreach(m -> fill!(m, zero(T)), eachstoredblock(data_matrix(ft)))
229229
return ft
230230
end
231231
Base.zeros(fta::FusionTensorAxes) = zeros(Float64, fta)
232232

233233
function Base.randn(rng::AbstractRNG, ::Type{T}, fta::FusionTensorAxes) where {T}
234234
ft = FusionTensor{T}(undef, fta)
235-
map(m -> randn!(rng, m), eachstoredblock(data_matrix(ft)))
235+
foreach(m -> randn!(rng, m), eachstoredblock(data_matrix(ft)))
236236
return ft
237237
end
238238
Base.randn(rng::AbstractRNG, fta::FusionTensorAxes) = randn(rng, Float64, fta)

src/permutedims/permutedims.jl

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,12 +46,11 @@ end
4646
function fusiontensor_permutedims!(
4747
new_ft::FusionTensor{T,N}, old_ft::FusionTensor{T,N}, flatperm::NTuple{N,Integer}
4848
) where {T,N}
49-
map(m -> fill!(m, zero(T)), eachstoredblock(data_matrix(new_ft)))
49+
foreach(m -> fill!(m, zero(T)), eachstoredblock(data_matrix(new_ft)))
5050
unitary = compute_unitary(new_ft, old_ft, flatperm)
51-
for p in unitary
52-
old_trees, new_trees = first(p)
51+
for ((old_trees, new_trees), coeff) in unitary
5352
new_block = view(new_ft, new_trees...)
5453
old_block = view(old_ft, old_trees...)
55-
@strided new_block .+= last(p) .* permutedims(old_block, flatperm)
54+
@strided new_block .+= coeff .* permutedims(old_block, flatperm)
5655
end
5756
end

0 commit comments

Comments
 (0)