batch_idx * len(data)

我认为这个len(data)完全就是batch_size,亲测就是的。(每组的数据量明显是batch_size)

猜你喜欢

转载自blog.csdn.net/weixin_42133481/article/details/114291322