Function discretize

models/mesh_xl/tokenizer.py:8–19

(
    t: Tensor,
    continuous_range: Tuple[float, float],
    num_discrete: int = 128
)

def discretize(
    t: Tensor,
    continuous_range: Tuple[float, float],
    num_discrete: int = 128
) -> Tensor:
    
    lo, hi = continuous_range
    assert hi > lo
    t = (t - lo) / (hi - lo)    # cube normalize
    t *= num_discrete
    t -= 0.5
    return t.round().long().clamp(min = 0, max = num_discrete - 1)