csukuangfj's picture
support batch size > 1
6037ea0