keras-js
keras-js copied to clipboard
Reshape layer with -1 throws assertion error
When using a model.json with a Reshape layer like the below code, the code here throws an exception because the reduce cannot calculate currectly.
I would love to submit a pull request to fix this if someone can guide me in the right direction.
{ "name": "reshape_5", "class_name": "Reshape", "config": { "name": "reshape_5", "trainable": true, "target_shape": [ -1, 12 ] }, "inbound_nodes": [ [ [ "conv2d_624", 0, 0, {} ] ] ] },