Tensorflow:定义类似conv2d的操作

时间:2017-06-19 18:47:30

标签: c++ tensorflow convolution

我想定义一个新的操作,它应该以{{1​​}}的工作方式工作。查看文档,我应该实现新操作类的tf.conv2d成员方法。据我所知,Compute(OpKernelContext* context)conv2d中定义。但是,文档展示了一个相当简单的运算符(conv_ops.cc)和ZeroOutOp有很多依赖关系。我无法找到有关如何执行实施的良好文档? 有什么想法,我可以找到有关tensorflow源代码的文档?任何教程都展示了如何实现相当复杂的操作?

0 个答案:

没有答案