backward.json

[
  {
    "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": "add",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
    "dim": 0,
    "fill_value": 0,
    "grad": [[10, 20], [15, 25]],
    "expected": [[10, 20], [15, 25], [15, 25], [10, 20]]
  },
  {
    "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": "mean",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
    "dim": 0,
    "fill_value": 0,
    "grad": [[10, 20], [15, 25]],
    "expected": [[10, 20], [15, 25], [15, 25], [10, 20]]
  },
  {
    "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]]
  },
  {
    "name": "max",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
    "dim": 0,
    "fill_value": 0,
    "grad": [[10, 20], [15, 25]],
    "expected": [[10, 0], [0, 25], [15, 0], [0, 20]]
  }
]