浏览代码

Update caffe.js (#618)

Lutz Roeder 5 年之前
父节点
当前提交
bcfb0f0e85
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      source/caffe.js

+ 3 - 3
source/caffe.js

@@ -616,12 +616,12 @@ caffe.Tensor = class {
             if (blob.channels != 1) {
                 shape.push(blob.channels);
             }
-            if (blob.width != 1) {
-                shape.push(blob.width);
-            }
             if (blob.height != 1) {
                 shape.push(blob.height);
             }
+            if (blob.width != 1) {
+                shape.push(blob.width);
+            }
         }
         else if (Object.prototype.hasOwnProperty.call(blob, 'shape')) {
             shape = blob.shape.dim;