Function load_tensor

tests/compare_mask_decoder.py:43–47

(path)

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, shape


def compare(name, ref_path, cpp_path):