Internal API

struct Factor{T, N}

Fields

• vars
• vals

Encodes a discrete function over the set of variables vars that maps each instantiation of vars into a nonnegative number in vals.

struct Samples{L} <: AbstractVector{SubArray{Float64, 1, Matrix{Float64}, Tuple{Base.Slice{Base.OneTo{Int64}}, Int64}, true}}

Fields

• samples::Matrix{Int64}

• labels::Vector

The sampled configurations are stored in samples, which is a vector of vector. labels is a vector of variable names for labeling configurations.

backward_tropical!(
    ixs,
    xs::Tuple,
    iy,
    y,
    ymask,
    size_dict
) -> Vector{Any}

The backward rule for tropical einsum.

• ixs and xs are labels and tensor data for input tensors,
• iy and y are labels and tensor data for the output tensor,
• ymask is the boolean mask for gradients,
• size_dict is a key-value map from tensor label to dimension size.

parse_mar_solution_file(
    rawlines::Vector{String};
    factor_eltype
) -> Vector{Vector{Float64}}

Parse the solution marginals of all variables from the UAI MAR solution file. The order of the variables is the same as in the model definition.

The UAI file formats are defined in: https://uaicompetition.github.io/uci-2022/file-formats/

read_query_file(
    query_filepath::AbstractString
) -> Vector{Int64}

Return the query variables in query_filepath. If the passed file path is an empty string, return an empty vector.

The UAI file formats are defined in: https://uaicompetition.github.io/uci-2022/file-formats/

rescale_array(tensor::AbstractArray{T}) -> RescaledArray

Returns a rescaled array that equivalent to the input tensor.

tensor_from_dict(dict::Dict)

Convert a dictionary back to a tensor.

Arguments

• dict::Dict: The dictionary representation of a tensor

Returns

• AbstractArray: The reconstructed tensor

Dictionary Structure Expected

• "size": The dimensions of the tensor
• "complex": Boolean indicating if the tensor contains complex numbers
• "data": The tensor data as a flat array of real numbers

tensor_to_dict(tensor::AbstractArray{T}) where T

Convert a tensor to a dictionary representation for JSON serialization.

Arguments

• tensor::AbstractArray{T}: The tensor to convert

Returns

• Dict: A dictionary containing tensor metadata and data

Dictionary Structure

• "size": The dimensions of the tensor
• "complex": Boolean indicating if the tensor contains complex numbers
• "data": The tensor data as a flat array of real numbers