modern-fortran / neural-fortran

A parallel framework for deep learning
MIT License
403 stars 83 forks source link

Implement a flatten layer #68

Closed milancurcic closed 2 years ago

milancurcic commented 2 years ago

Which will allow chaining conv2d and maxpool2d layers with a dense layer.