search

lgarithm / crystalnet

crystalnet -- a mini core AI library (being refactored, see https://github.com/lgarithm/stdnn-ops)
MIT License
16 stars 3 forks source link

Generic tensor iterator #91

Open lgarithm opened 6 years ago

lgarithm commented 6 years ago

It should be memory efficient. The element of each iteration should be exactly the same shape, i.e. dynamic batch size will not be supported ATM.