(path: str, t: torch.Tensor)
| 22 | |
| 23 | |
| 24 | def save_tensor(path: str, t: torch.Tensor) -> None: |
| 25 | t = t.detach().cpu().float().contiguous() |
| 26 | os.makedirs(os.path.dirname(path) or ".", exist_ok=True) |
| 27 | with open(path + ".bin", "wb") as f: |
| 28 | f.write(t.numpy().tobytes()) |
| 29 | with open(path + ".shape", "w") as f: |
| 30 | f.write(",".join(str(d) for d in t.shape)) |
| 31 | |
| 32 | |
| 33 | def save_i32_tensor(path: str, values: List[int]) -> None: |