Summary: 3 instances, 3 unique Text Count compute_per_sample = True #TODO: better than manually set 1 mask = dist < distances #TODO check that broadcasting of above is correct 1 # TODO: this should also be done with the ProgressMeter 1