We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ff7a6d0 commit 11e1761Copy full SHA for 11e1761
src/caches/beliefpropagationcache.jl
@@ -16,11 +16,11 @@ using NDTensors: NDTensors
16
17
default_message(elt, inds_e) = ITensor[denseblocks(delta(elt, i)) for i in inds_e]
18
default_messages(ptn::PartitionedGraph) = Dictionary()
19
-function default_message_update(contract_list::Vector{ITensor}; kwargs...)
+function default_message_update(contract_list::Vector{ITensor}; normalize=true, kwargs...)
20
sequence = optimal_contraction_sequence(contract_list)
21
updated_messages = contract(contract_list; sequence, kwargs...)
22
message_norm = norm(updated_messages)
23
- if !iszero(message_norm)
+ if normalize && !iszero(message_norm)
24
updated_messages /= message_norm
25
end
26
return ITensor[updated_messages]
0 commit comments