Method Batch

mlx/data/stream/Batch.cpp:11–23

namespace data {
namespace stream {
Batch::Batch(
    const std::shared_ptr<Stream>& stream,
    int64_t batch_size,
    const std::unordered_map<std::string, double>& pad_values,
    const std::unordered_map<std::string, int>& batch_dims)
    : stream_(stream),
      batchSize_(batch_size),
      padValues_(pad_values),
      batchDims_(batch_dims) {
  if (batch_size <= 0) {
    throw std::runtime_error("Batch: batch size must be positive");
  }
}

Sample Batch::next() const {
  std::vector<Sample> samples;