Method next

mlx/data/stream/Batch.cpp:25–39

}

Sample Batch::next() const {
  std::vector<Sample> samples;
  for (int i = 0; i < batchSize_; i++) {
    auto sample = stream_->next();
    if (sample.empty()) {
      break;
    }
    samples.push_back(std::move(sample));
  }
  if (samples.empty()) {
    return Sample();
  } else {
    return core::merge_batch(samples, padValues_, batchDims_);
  }
}

void Batch::reset() {
  stream_->reset();