Lutz Roeder 5 лет назад
Родитель
Сommit
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;