Tensorflow:如何从另一个操作中调用C ++中的一个操作

时间:2018-06-29 10:27:13

标签: c++ tensorflow

我正在用C ++编写一个新操作,并且想利用另一个操作。例如,我构造了两个新的张量,并希望在操作中进行矩阵乘法,因此我想使用MatMulOp。但是,MatMulOp::Compute需要输入OpKernelContext

我的问题是如何形成这样的输入来调用此操作?

0 个答案:

没有答案
相关问题