{"train": [{"input": [[5, 0, 0], [0, 5, 0], [5, 0, 0]], "output": [[2, 2, 2], [4, 4, 4], [2, 2, 2]]}, {"input": [[0, 0, 5], [0, 0, 5], [0, 0, 5]], "output": [[3, 3, 3], [3, 3, 3], [3, 3, 3]]}, {"input": [[0, 5, 0], [0, 0, 5], [0, 5, 0]], "output": [[4, 4, 4], [3, 3, 3], [4, 4, 4]]}, {"input": [[0, 0, 5], [0, 5, 0], [5, 0, 0]], "output": [[3, 3, 3], [4, 4, 4], [2, 2, 2]]}], "test": [{"input": [[0, 0, 5], [5, 0, 0], [0, 5, 0]], "output": [[3, 3, 3], [2, 2, 2], [4, 4, 4]]}]}