c++ tensorflow random_normal实现

在 C++ 中使用 TensorFlow 实现 random_normal 的方法如下：

#include <tensorflow/core/public/session.h>
#include <tensorflow/core/framework/tensor.h>
#include <tensorflow/core/kernels/random_op.h>

// 定义函数
tensorflow::Tensor random_normal(tensorflow::Session* session, const std::vector<int64_t>& shape) {
    // 创建 Op
    tensorflow::ops::RandomStandardNormal op = tensorflow::ops::RandomStandardNormal::Builder()
        .Seed(0)
        .Build();
    // 执行 Op
    std::vector<tensorflow::Tensor> outputs;
    TF_CHECK_OK(session->Run({}, {"random_normal"}, {}, &outputs));
    // 转换结果
    tensorflow::Tensor result = outputs[0];
    result = tensorflow::ops::Reshape(result, shape).output;
    return result;
}

// 示例
int main() {
    tensorflow::Session* session;
    tensorflow::SessionOptions options;
    tensorflow::Status status = tensorflow::NewSession(options, &session);
    if (!status.ok()) {
        std::cerr << "Error creating TensorFlow session: " << status.ToString() << std::endl;
        return 1;
    }
    tensorflow::Tensor input = random_normal(session, {2, 3});
    std::cout << input.DebugString();
    session->Close();
    return 0;
}

在上面的示例中，我们调用了 `random_normal` 函数来生成一个指定形状的张量。该函数首先创建了一个 `RandomStandardNormal` 操作，然后使用 `session->Run` 方法执行该操作，并将结果转换为指定形状的张量。最后，我们使用 `DebugString` 方法打印出了生成的张量。