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": "add",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
    "dim": 0,
    "fill_value": 0,
    "expected": [[6, 5], [6, 8]]
  },
  {
    "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": 9,
    "expected": [[9, 9, 5, 6, 6, 9], [7, 5, 5, 9, 9, 9]]
  },
  {
    "name": "sub",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
    "dim": 0,
    "fill_value": 9,
    "expected": [[3, 4], [3, 1]]
  },
  {
    "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": "mul",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
    "dim": 0,
    "fill_value": 1,
    "expected": [[5, 6], [8, 15]]
  },
  {
    "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": "div",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[4, 2], [2, 1], [4, 2], [1, 2]],
    "dim": 0,
    "fill_value": 1,
    "expected": [[0.25, 0.25], [0.125, 0.5]]
  },
  {
    "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": "mean",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
    "dim": 0,
    "fill_value": 0,
    "expected": [[3, 2.5], [3, 4]]
  },
  {
    "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": "max",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
    "dim": 0,
    "fill_value": 0,
    "expected": [[5, 3], [4, 5]],
    "expected_arg": [[0, 3], [2, 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]]
  },
  {
    "name": "min",
    "index": [[0, 0], [1, 1], [1, 1], [0, 0]],
    "input": [[5, 2], [2, 5], [4, 3], [1, 3]],
    "dim": 0,
    "fill_value": 9,
    "expected": [[1, 2], [2, 3]],
    "expected_arg": [[3, 0], [1, 2]]
  }
]