-
8 votesanswersviews
为什么选择我的自定义操作系统的CPU实现?
为了学习如何编写自定义TensorFlow操作,我按照Adding a New Op教程制作了一个"add_b"操作,为每个输入值添加一个标量 b . add_b_op.cc : #define EIGEN_USE_THREADS #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" ... -
2 votesanswersviews
使用在张量流中具有变量依赖性的自定义op替换图中的节点
我试图用自定义操作替换图中完成的计算 . 假设图表有一个常量 A 和权重变量 W ,我创建自定义操作来获取这两个输入并执行整个计算(除了重量更新的最后一步): custom_op_tensor = custom_module.custom_op([A,W]) g_def = tf.get_default_graph().as_graph_def() input_map = { tensor.nam...