Method squeeze

mlx/data/Array.cpp:136–165

}

void Array::squeeze(const std::vector<int>& dims) {
  std::vector<int64_t> mask(ndim(), true);
  if (dims.empty()) {
    for (int i = 0; i < ndim(); i++) {
      if (shape_[i] == 1) {
        mask[i] = false;
      }
    }
  } else {
    for (auto dim : dims) {
      dim = checkdim(dim);
      if (shape_[dim] == 1) {
        mask[dim] = false;
      } else {
        throw std::runtime_error(
            "Array: cannot squeeze a non-singleton dimension");
      }
    }
  }
  int newndim = 0;
  for (int i = 0; i < shape_.size(); i++) {
    if (mask[i]) {
      if (i != newndim) {
        shape_[newndim] = shape_[i];
      }
      newndim++;
    }
  }
  shape_.resize(newndim);
}

void Array::reshape(const std::vector<int64_t>& shape) {
  std::vector<int64_t> new_shape = shape;