[
  {
    "name": "add",
    "index": [[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]],
    "input": [[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]],
    "dim": 1,
    "fill_value": 0,
    "grad": [[10, 20, 30, 40, 50, 60], [15, 25, 35, 45, 55, 65]],
    "expected": [[50, 60, 50, 30, 40], [15, 15, 35, 35, 25]]
  },
  {
    "name": "mean",
    "index": [[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]],
    "input": [[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]],
    "dim": 1,
    "fill_value": 0,
    "grad": [[10, 20, 30, 40, 50, 60], [15, 25, 35, 45, 55, 65]],
    "expected": [[50, 60, 50, 30, 40], [15, 15, 35, 35, 25]]
  },
  {
    "name": "max",
    "index": [[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]],
    "input": [[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]],
    "dim": 1,
    "fill_value": 0,
    "grad": [[10, 20, 30, 40, 50, 60], [15, 25, 35, 45, 55, 65]],
    "expected": [[50, 60, 0, 30, 40], [0, 15, 0, 35, 25]]
  }
]