MolecularDiffusion.modules.models.tabasco.flow.utils

Classes

HistogramTimeDistribution

Module Contents

class MolecularDiffusion.modules.models.tabasco.flow.utils.HistogramTimeDistribution(probs: torch.Tensor)

sample(sample_shape: torch.Size = torch.Size([])) → torch.Tensor

bin_width

categorical

probs

uniform