Fix tensor selection inside contract

bsc-quantic · Dec 4, 2023 · 7bf04a6 · 7bf04a6
1 parent 2f77210
commit 7bf04a6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/TensorNetwork.jl b/src/TensorNetwork.jl
@@ -434,8 +434,7 @@ The `kwargs` will be passed down to the [`einexpr`](@ref) function.
 See also: [`einexpr`](@ref), [`contract!`](@ref).
 """
 function contract(tn::AbstractTensorNetwork; path = einexpr(tn))
- # TODO does `first` work always?
- length(path.args) == 0 && return select(tn, inds(path)) |> first
+ length(path.args) == 0 && return tn[inds(path)...]
 
  intermediates = map(subpath -> contract(tn; path = subpath), path.args)
  contract(intermediates...; dims = suminds(path))