Kaynağa Gözat

Update nnabla-proto.js

Lutz Roeder 2 yıl önce
ebeveyn
işleme
b4bc5910c8
3 değiştirilmiş dosya ile 163 ekleme ve 1 silme
  1. 159 0
      source/nnabla-metadata.json
  2. 0 0
      source/nnabla-proto.js
  3. 4 1
      tools/nnabla_script.py

+ 159 - 0
source/nnabla-metadata.json

@@ -7791,5 +7791,164 @@
         "description": "N-D array with shape :math:`(N, C_y, C_x, H_o, W_o)`.\n\nA spatial size of the output is calculated as\n\n.. math::\n\n  H_o = \\frac{H + (top\\_pad + bottom\\_pad) - patch_v }{patch\\_step_v} + 1.\n\nA channel size of the output is calculated as\n\n.. math::\n\n  C_y = \\frac{2 \\times shift_v}{shift\\_step_v} + 1.\n\n:math:`W_o` and :math:`C_x` are the same calculation with differenct components."
       }
     ]
+  },
+  {
+    "name": "Unique",
+    "description": "Find the unique elements of input array.",
+    "inputs": [
+      {
+        "name": "x",
+        "type": "nnabla.Variable",
+        "description": "A N-D array."
+      }
+    ],
+    "attributes": [
+      {
+        "name": "flatten",
+        "type": "boolean",
+        "default": true,
+        "description": "If True, unique values of the flatten input array are returned."
+      },
+      {
+        "name": "axis",
+        "type": "int64",
+        "default": "None",
+        "description": "If flatten is True and axis is specified, unique slices along axis are returned."
+      },
+      {
+        "name": "sorted",
+        "type": "boolean",
+        "default": true,
+        "description": "If True, unique values/slices sorted in ascending order are returned."
+      },
+      {
+        "name": "with_index",
+        "type": "boolean",
+        "default": false,
+        "description": "If True, `indices` is returned."
+      },
+      {
+        "name": "with_inverse",
+        "type": "boolean",
+        "default": false,
+        "description": "If True, `inverse_indices` is returned."
+      },
+      {
+        "name": "with_counts",
+        "type": "boolean",
+        "default": false,
+        "description": "If True, `counts` is returned."
+      }
+    ],
+    "outputs": [
+      {
+        "name": "y",
+        "type": "nnabla.Variable",
+        "description": "A N-D array."
+      },
+      {
+        "name": "indices",
+        "type": "nnabla.Variable",
+        "description": "A 1-D array. It's indices of `y` elements first occurance in `x`. If `flatten` is True, it contains indices to flattend input array `x`. If `flatten` is False and `axis` is specified, it contains indices to input array `x` on `axis`."
+      },
+      {
+        "name": "inverse_indices",
+        "type": "nnabla.Variable",
+        "description": "A 1-D array. It's indices of `x` elements corresponding to `y`. If `flatten` is True, it contains indices to output array `y`. If `flatten` is False and `axis` is specified, it contains indices to output array `y` on `axis`."
+      },
+      {
+        "name": "counts",
+        "type": "nnabla.Variable",
+        "description": "A 1-D array. It's the count of each element of 'y' in input array `x`."
+      }
+    ]
+  },
+  {
+    "name": "EyeLike",
+    "description": "Generate a 2-D array with ones on the diagonal, specified by `k`, and zeros elsewhere.\nThe shape of the output array is the same as the input array.",
+    "inputs": [
+      {
+        "name": "x",
+        "type": "nnabla.Variable",
+        "description": "A 2-D array."
+      }
+    ],
+    "attributes": [
+      {
+        "name": "k",
+        "type": "int64",
+        "default": 0,
+        "description": "Index of the diagonal. The default value 0 means the main diagonal, a positive value means an upper diagonal, and a negative value means a lower diagonal."
+      }
+    ],
+    "outputs": [
+      {
+        "name": "y",
+        "type": "nnabla.Variable",
+        "description": "A 2-D array."
+      }
+    ]
+  },
+  {
+    "name": "Mod2",
+    "description": "Element-wise remainder function.\nThe behavior of this opeator is determined by x0's dtype and the `fmod` argument:\n\n.. math::\n    y_i = \\left\\{\n    \\begin{array}{ll}\n      \\text{numpy.fmod}(x_{0,i}, x_{1,i})\n        & (x_{0} \\text{has a floating-point type or fmod is True})\\\\\n      \\text{numpy.mod}(x_{0,i}, x_{1,i})\n        & (\\text{otherwise})\n    \\end{array} \\right..",
+    "inputs": [
+      {
+        "name": "x0",
+        "type": "nnabla.Variable",
+        "description": "A N-D array."
+      },
+      {
+        "name": "x1",
+        "type": "nnabla.Variable",
+        "description": "A N-D array."
+      }
+    ],
+    "attributes": [
+      {
+        "name": "fmod",
+        "type": "boolean",
+        "default": false,
+        "description": "If True, this operator behaves like numpy.fmod, otherwise it behaves like numpy.mod."
+      }
+    ],
+    "outputs": [
+      {
+        "name": "y",
+        "type": "nnabla.Variable",
+        "description": "A N-D array."
+      }
+    ]
+  },
+  {
+    "name": "BitShift",
+    "description": "Element-wise bit shift function.",
+    "inputs": [
+      {
+        "name": "x",
+        "type": "nnabla.Variable",
+        "description": "A N-D array. Its dtype must be one of the unsigned integer types."
+      },
+      {
+        "name": "shift",
+        "type": "nnabla.Variable",
+        "description": "A N-D array. Its dtype is casted to x's dtype at run-time."
+      }
+    ],
+    "attributes": [
+      {
+        "name": "direction",
+        "type": "string",
+        "default": "LEFT",
+        "description": "Direction of bit shift."
+      }
+    ],
+    "outputs": [
+      {
+        "name": "y",
+        "type": "nnabla.Variable",
+        "description": "A N-D array."
+      }
+    ]
   }
 ]

Dosya farkı çok büyük olduğundan ihmal edildi
+ 0 - 0
source/nnabla-proto.js


+ 4 - 1
tools/nnabla_script.py

@@ -121,7 +121,10 @@ def _attribute(name, value): # pylint: disable=too-many-branches
         attribute['type'] = 'int64'
         if default:
             _ = value['default']
-            attribute['default'] = int(_) if isinstance(_, str) and not _.startswith('len') else _
+            if isinstance(_, str) and not _.startswith('len') and _ != 'None':
+                attribute['default'] = int(_)
+            else:
+                attribute['default'] = _
     elif value['type'] == 'repeated int64':
         attribute['type'] = 'int64[]'
     elif value['type'] == 'repeated float':

Bu fark içinde çok fazla dosya değişikliği olduğu için bazı dosyalar gösterilmiyor