gpt4 book ai didi

tensorflow - 理解 C++ 中新 Tensorflow 运算符的定义

转载 作者:行者123 更新时间:2023-12-04 01:04:26 27 4
gpt4 key购买 nike

我正在尝试遵循在 tensorflow 中定义新运算符的官方指南。 https://www.tensorflow.org/extend/adding_an_op

#include "tensorflow/core/framework/op.h"
#include "tensorflow/core/framework/shape_inference.h"

using namespace tensorflow;

REGISTER_OP("ZeroOut")
.Input("to_zero: int32")
.Output("zeroed: int32")
.SetShapeFn([](::tensorflow::shape_inference::InferenceContext* c){
c->set_output(0, c->input(0));
return Status::OK();
});

但是我找不到这段代码的逐行解释,特别是我不明白 .SetShapeFn([](::tensorflow::shape_inference::InferenceContext* c) 的作用是什么及其语法。另外我我对 InferenceContext 感到困惑,我猜这是一种连续传递任何数组元素的方法.. 我在任何地方都找不到明确的定义,也许我找错了地方,有人可以帮助我解释或引用?
我想深入了解这段代码在幕后做了什么。

最佳答案

您是否在此处发现了有关形状推理函数的部分?
https://www.tensorflow.org/extend/adding_an_op#shape_functions_in_c

这对 ShapeInferenceContext 类和编写自己的函数的机制有很多讨论。如果这不包括您感兴趣的内容,您能否提供更多详细信息?

关于tensorflow - 理解 C++ 中新 Tensorflow 运算符的定义,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46152256/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com