neuroailab / tfutils

Utilities for working with tensorflow
MIT License
25 stars 8 forks source link

Changes to do enqueue of batch_size of 1 #75

Closed chengxuz closed 7 years ago

chengxuz commented 7 years ago

For enqueue rather than enqueue_many, the first dimension of the shape arguments sent into the queue initialization function should not be excluded. (One also need to do shape setting and squeezing in their data_providers for the first dimension)