Function save_tensor

tests/dump_reference.py:27–34

Save a tensor as raw float32 binary + shape file.

(path, t)

def save_tensor(path, t):
    """Save a tensor as raw float32 binary + shape file."""
    t = t.detach().cpu().float().contiguous()
    with open(path + ".bin", "wb") as f:
        f.write(t.numpy().tobytes())
    with open(path + ".shape", "w") as f:
        f.write(",".join(str(d) for d in t.shape))
    print(f"  saved {path} shape={list(t.shape)} bytes={t.numel()*4}")


def main():