소스 검색

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;