majianjia / nnom

A higher-level Neural Network library for microcontrollers.

Apache License 2.0

908 stars 245 forks source link

fix reshape output shape error. #193

Closed songdaw closed 1 year ago

songdaw commented 1 year ago

Reshape output shape error when run on 32bit cpu (eg. arm cortex-m). For example, without this commit, layer info: ......

9 Dense - HrdTanH - ( 96, ) 6144 ( 64, 96, 128)

10 Reshape - - ( 1, 3, 7,) ( 96, 21, 0)

11 Concat - - ( 1, 3, 39,) ( 117, 117, 0)

...... with this commit, layer info: ......

9 Dense - HrdTanH - ( 96, ) 6144 ( 64, 96, 128)

10 Reshape - - ( 1, 3, 32,) ( 96, 96, 0)

11 Concat - - ( 1, 3, 64,) ( 192, 192, 0)

....