Function load_tensor

tests/check_preprocessing.py:6–11

(path)

import sys

def load_tensor(path):
    with open(path + ".shape") as f:
        shape = [int(x) for x in f.read().strip().split(",")]
    with open(path + ".bin", "rb") as f:
        data = np.frombuffer(f.read(), dtype=np.float32).copy()
    return data.reshape(shape)

ref_dir = sys.argv[1] if len(sys.argv) > 1 else "tests/ref_phase3"