在张量流中使用input_fn时如何设置列批量大小

Question

在张量流estimator/estimator.py，函数fit，partial_fit，predict等中，它表示在使用input_fn时，batch_size必须没有。 那么，在哪里设置batch_size。在我的代码中，需要input_fn，因为我用它来从原始数据中选择一些特征。 感谢。