Merge pull request #132 from changy1105/reduce_mean

feat(paddlejs-backend-webgl): add reduce_mean op

Author: yueshuangyan

packages/paddlejs-backend-webgl/src/ops/index.ts

@@ -40,6 +40,7 @@ import where from './shader/where';
 import connect from './shader/connect';
 import squeeze2 from './shader/squeeze2';
 import pad3d from './shader/pad3d';
+import reduce_mean from './shader/reduce_mean';
 
 const ops = {
     arg_max,
@@ -76,6 +77,7 @@ const ops = {
     reduce_sum,
     where,
     connect,
+    reduce_mean,
     prelu: dynamic('prelu'),
     relu6: dynamic('relu6'),
     leakyRelu: dynamic('leakyRelu'),

packages/paddlejs-backend-webgl/src/ops/shader/reduce_mean.ts

@@ -0,0 +1,37 @@
+
+/**
+ * @file reduce_mean
+ */
+
+function mainFunc(
+    {},
+    { inputs_dim, dim }
+) {
+    return `
+    // start函数
+    void main(void) {
+        ivec4 oPos = getOutputTensorPos();
+        // 输出坐标转换为输入坐标
+        float o = 0.0;
+        for (int i = 0; i < ${inputs_dim}; i++) {
+            oPos[${dim}] = i;
+            o += getValueFromTensorPos_origin(oPos.r, oPos.g, oPos.b, oPos.a);
+        }
+        o = o / float(${inputs_dim});
+        setOutput(o);
+    }
+    `;
+}
+export default {
+    mainFunc,
+    params: [
+        'inputs_dim',
+        'dim'
+    ],
+    textureFuncConf: {
+        origin: ['getValueFromTensorPos']
+    },
+    behaviors: [
+        'normalizeDim'
+    ]
+};

packages/paddlejs-backend-webgl/test/op/data/reduce_mean.json

@@ -0,0 +1,60 @@
+{
+  "ops": [
+    {
+      "attrs": {
+        "op_device": ""
+      },
+      "inputs": {
+        "X": [
+          "feed"
+        ]
+      },
+      "outputs": {
+        "Out": [
+          "mean_0.tmp_0"
+        ]
+      },
+      "type": "feed"
+    },
+    {
+      "attrs": {
+        "axis": 0
+      },
+      "inputs": {
+        "X": ["mean_0.tmp_0"]
+      },
+      "outputs": {
+        "Out": ["mean_0.tmp_out"]
+      },
+      "type": "reduce_mean"
+    },
+    {
+      "attrs": {
+        "op_device": ""
+      },
+      "inputs": {
+        "X": [
+          "mean_0.tmp_out"
+        ]
+      },
+      "outputs": {
+        "Out": [
+          "fetch"
+        ]
+      },
+      "type": "fetch"
+    }
+  ],
+  "vars": [
+    {
+      "data": [1, 2, 3, 4, 5, 6, 7, 8],
+      "name": "mean_0.tmp_0",
+      "shape": [2, 4]
+    },
+    {
+      "data":[3, 4, 5, 6],
+      "name": "mean_0.tmp_out",
+      "shape": [1, 4]
+    }
+  ]
+}