Function load_tensor

tests/dump_cpp_preprocessed.py:33–37

(path)

    return chw

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

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