Function makeTensor

tensorflow/core/util/test_util.h:36–43

template<typename T>
Tensor makeTensor(std::vector<long long int> tensorShape, DataType dataType, float min_val, float max_val) {
    auto outputTensor = Tensor(dataType, TensorShape(tensorflow::gtl::ArraySlice<long long int>(tensorShape.data(), tensorShape.size())));
    auto range = max_val - min_val;
    outputTensor.flat<T>() = outputTensor.flat<T>().template setRandom<Eigen::internal::UniformRandomGenerator<T>>(); // (0, 1)
    outputTensor.flat<T>() = outputTensor.flat<T>() * outputTensor.flat<T>().constant(static_cast<T>(range)); // (0, range)
    outputTensor.flat<T>() = outputTensor.flat<T>() + outputTensor.flat<T>().constant(static_cast<T>(min_val)); // (min_val, max_val)
    return outputTensor;
}

inline Tensor makeTensor(std::vector<long long int> tensorShape, DataType dataType, float min_val, float max_val) {
        switch(dataType) {