Fatal编程技术网
  • cplusplus/
  • C++ Tensorflow：定义类似conv2d的操作

C++ Tensorflow：定义类似conv2d的操作

c++ tensorflow

C++ Tensorflow：定义类似conv2d的操作,c++,tensorflow,convolution,C++,Tensorflow,Convolution,我想定义一个新的操作，它的工作方式应该与tf.conv2d的工作方式相同。查看文档，我应该实现新操作类的Compute（OpKernelContext*context）member方法。据我所知，conv2d是在conv_ops.cc中定义的。然而，文档显示了一个非常简单的运算符（zerootop），并且conv_ops.cc有很多依赖项。我找不到关于如何实施的好文档？ 在哪里可以找到tensorflow源代码的文档？有介绍如何实现相当复杂的操作的教程吗

我想定义一个新的操作，它的工作方式应该与
tf.conv2d
的工作方式相同。查看文档，我应该实现新操作类的
Compute（OpKernelContext*context）
member方法。据我所知，
conv2d
是在
conv_ops.cc
中定义的。然而，文档显示了一个非常简单的运算符（
zerootop
），并且
conv_ops.cc
有很多依赖项。我找不到关于如何实施的好文档？ 在哪里可以找到tensorflow源代码的文档？有介绍如何实现相当复杂的操作的教程吗