Function load_shape

tests/dump_phase7_reference.py:37–39

(path: str)

def load_shape(path: str) -> List[int]:
    with open(path + ".shape", "r", encoding="utf-8") as f:
        return [int(x) for x in f.read().strip().split(",") if x]


def load_tensor(path: str) -> torch.Tensor: