微信扫一扫
一开始我还在嘲笑,为啥作者的代码要利用这个shape信息来动态地获取batch_size。
回答很简单,emmm,因为在最后一个batch中,数据可能不足一个设定的batch_size。
相关推荐