[numpy]array_split

  • sorted_indices分割成长度为num_sections的小片段。
self.batches = numpy.array_split(sorted_indices, num_sections)

推荐阅读更多精彩内容