torch_mist.utils.data.sampler

Module Contents

Classes

SameAttributeSampler

class torch_mist.utils.data.sampler.SameAttributeSampler(batch_size: int, neg_samples: int, attributes: torch.Tensor | numpy.ndarray)

Bases: torch.utils.data.Sampler

__iter__()

__len__()