C++实现python-linspace函数功能

#include 
#include 

template
std::vector linspace(T start_in, T end_in, int num_in)
{

  std::vector linspaced;

  double start = static_cast(start_in);
  double end = static_cast(end_in);
  double num = static_cast(num_in);

  if (num == 0) { return linspaced; }
  if (num == 1) 
    {
      linspaced.push_back(start);
      return linspaced;
    }

  double delta = (end - start) / (num - 1);

  for(int i=0; i < num-1; ++i)
    {
      linspaced.push_back(start + delta * i);
    }
  linspaced.push_back(end); // I want to ensure that start and end
                            // are exactly the same as the input
  return linspaced;
}

void print_vector(std::vector vec)
{
  std::cout << "size: " << vec.size() << std::endl;
  for (double d : vec)
    std::cout << d << " ";
  std::cout << std::endl;
}

int main()
{
  std::vector vec_1 = linspace(1, 10, 3);
  print_vector(vec_1);

  std::vector vec_2 = linspace(6.0, 23.4, 5);
  print_vector(vec_2);

  std::vector vec_3 = linspace(0.0, 2.0, 1);
  print_vector(vec_3);

  std::vector vec_4 = linspace(0.0, 2.0, 0);
  print_vector(vec_4);


  return 0;
}

来自：
lhttps://www.cnblogs.com/LuckCoder/p/16381130.html