gflownet.utils.molecule.distributions

Functions

get_mixture_of_projected_normals(weights, concentrations)

Module Contents

gflownet.utils.molecule.distributions.get_mixture_of_projected_normals(weights, concentrations)

Parameters:

• weights – torch.tensor of shape [*batch_shape, n_components]

• concentrations – torch.tensor of shape [*batch_shape, n_components, n_dim]