added backward test

d66dbcce · rusty1s · 2ba6da8a · d66dbcce
Commit d66dbcce authored Dec 28, 2017 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

test/backward.json test/backward.json +9 -0

No files found.
--- a/test/backward.json
+++ b/test/backward.json
@@ -8,6 +8,15 @@
    "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]],