-
由 yoonliu 提交于
After flatten layer, the input dims 4 should be squeezed to 2. The original bug with output dims 4 is corrected.
38ef8d46
After flatten layer, the input dims 4 should be squeezed to 2. The original bug with output dims 4 is corrected.