Function MakeLinspaceArray2D

tensorflow/compiler/xla/array2d.h:101–116

// with dimensions n1 x n2.
template <typename NativeT = float>
std::unique_ptr<Array2D<NativeT>> MakeLinspaceArray2D(double from, double to,
                                                      int64 n1, int64 n2) {
  auto array = absl::make_unique<Array2D<NativeT>>(n1, n2);
  int64 count = n1 * n2;
  NativeT step =
      static_cast<NativeT>((count > 1) ? (to - from) / (count - 1) : 0);
  auto set = [&array, n2](int64 index, NativeT value) {
    (*array)(index / n2, index % n2) = value;
  };
  for (int64 i = 0; i < count - 1; ++i) {
    set(i, (static_cast<NativeT>(from) +
            static_cast<NativeT>(i) * static_cast<NativeT>(step)));
  }
  set(count - 1, static_cast<NativeT>(to));
  return array;
}
}  // namespace xla

#endif  // TENSORFLOW_COMPILER_XLA_ARRAY2D_H_