Skip to content

Commit 11e1761

Browse files
authored
Add normalize flag in default message update (#195)
1 parent ff7a6d0 commit 11e1761

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/caches/beliefpropagationcache.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,11 @@ using NDTensors: NDTensors
1616

1717
default_message(elt, inds_e) = ITensor[denseblocks(delta(elt, i)) for i in inds_e]
1818
default_messages(ptn::PartitionedGraph) = Dictionary()
19-
function default_message_update(contract_list::Vector{ITensor}; kwargs...)
19+
function default_message_update(contract_list::Vector{ITensor}; normalize=true, kwargs...)
2020
sequence = optimal_contraction_sequence(contract_list)
2121
updated_messages = contract(contract_list; sequence, kwargs...)
2222
message_norm = norm(updated_messages)
23-
if !iszero(message_norm)
23+
if normalize && !iszero(message_norm)
2424
updated_messages /= message_norm
2525
end
2626
return ITensor[updated_messages]

0 commit comments

Comments
 (0)