Function load_tensor

tests/dump_phase6_reference.py:37–41

(path: str)

def load_tensor(path: str) -> torch.Tensor:
    with open(path + ".shape", "r", encoding="utf-8") 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)


def save_raw(path: str, arr: np.ndarray, shape) -> None: