Method size

mlx/data/core/BatchShape.cpp:14–20

BatchShape::BatchShape(int dim) : dim_(dim), nodim_(false), num_sample_(0) {};

int64_t BatchShape::size() const {
  int64_t size = 1;
  for (auto dim : shape_) {
    size *= dim;
  }
  return size;
}

const std::vector<int64_t>& BatchShape::shape() const {
  return shape_;