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