Function save_raw

tests/dump_phase7_reference.py:67–72

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

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_tensor(path: str, t: torch.Tensor) -> None: