Function load_tensor

tests/dump_geom_encoder_reference.py:82–86

(path: str)

def load_tensor(path: str) -> torch.Tensor:
    with open(path + ".shape", "r") as f:
        shape = [int(x) for x in f.read().strip().split(",") if x]
    data = np.fromfile(path + ".bin", dtype=np.float32).reshape(shape)
    return torch.from_numpy(data)


# ── Sinusoidal position encoding (matches PositionEmbeddingSine) ──────────