jaxdem.analysis.pairs#

Functions

build_pairs(binspec)

Build (pair_i, pair_j, bin_id) arrays from a BinSpec.

flatten_pairs(binspec)

Deprecated alias for build_pairs().

Classes

FlatPairs

Pairs(pair_i, pair_j, bin_id, counts_per_bin)

Flat representation of bin tuples, suitable for JAX execution.
class jaxdem.analysis.pairs.Pairs(pair_i: ndarray, pair_j: ndarray, bin_id: ndarray, counts_per_bin: ndarray)[source]#

Bases: object

Flat representation of bin tuples, suitable for JAX execution.

pair_i#

shape (P,) int array

Type:

numpy.ndarray

pair_j#

shape (P,) int array

Type:

numpy.ndarray

bin_id#

shape (P,) int array in [0, B)

Type:

numpy.ndarray

counts_per_bin#

shape (B,) int array (number of tuples per bin)

Type:

numpy.ndarray

pair_i: ndarray#
pair_j: ndarray#
bin_id: ndarray#
counts_per_bin: ndarray#
jaxdem.analysis.pairs.build_pairs(binspec: BinSpec) → Pairs[source][source]#

Build (pair_i, pair_j, bin_id) arrays from a BinSpec.

jaxdem.analysis.pairs.flatten_pairs(binspec: BinSpec) → Pairs[source][source]#

Deprecated alias for build_pairs().

jaxdem.analysis.pairs.FlatPairs#

alias of Pairs