forward.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,
    "expected": [[0, 0, 4, 3, 3, 0], [2, 4, 4, 0, 0, 0]]
  },
  {
    "name": "sub",
    "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,
    "expected": [[0, 0, -4, -3, -3, 0], [-2, -4, -4, 0, 0, 0]]
  },
  {
    "name": "mul",
    "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": 1,
    "expected": [[1, 1, 4, 3, 2, 0], [0, 4, 3, 1, 1, 1]]
  },
  {
    "name": "div",
    "index": [[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]],
    "input": [[2, 1, 1, 4, 2], [1, 2, 1, 2, 4]],
    "dim": 1,
    "fill_value": 1,
    "expected": [[1, 1, 0.25, 0.5, 0.5, 1], [0.5, 0.25, 0.5, 1, 1, 1]]
  },
  {
    "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,
    "expected": [[0, 0, 4, 3, 1.5, 0], [1, 4, 2, 0, 0, 0]]
  },
  {
    "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,
    "expected": [[0, 0, 4, 3, 2, 0], [2, 4, 3, 0, 0, 0]],
    "expected_arg": [[-1, -1, 3, 4, 0, 1], [1, 4, 3, -1, -1, -1]]
  },
  {
    "name": "min",
    "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": 9,
    "expected": [[9, 9, 4, 3, 1, 0], [0, 4, 1, 9, 9, 9]],
    "expected_arg": [[-1, -1, 3, 4, 2, 1], [0, 4, 2, -1, -1, -1]]
  }
]