Function save_raw

tests/dump_geom_encoder_reference.py:41–46

(path: str, arr: np.ndarray, shape)

# ── Tensor save utilities ───────────────────────────────────────────────────

def save_raw(path: str, arr: np.ndarray, shape) -> None:
    os.makedirs(os.path.dirname(path) or ".", exist_ok=True)
    with open(path + ".bin", "wb") as f:
        f.write(arr.astype(np.float32, copy=False).tobytes())
    with open(path + ".shape", "w", encoding="utf-8") as f:
        f.write(",".join(str(int(d)) for d in shape))


def save_ggml_bnd(path: str, x_bnd: torch.Tensor) -> None: