标签: c++ tensorflow
我正在用C ++编写一个新操作,并且想利用另一个操作。例如,我构造了两个新的张量,并希望在操作中进行矩阵乘法,因此我想使用MatMulOp。但是,MatMulOp::Compute需要输入OpKernelContext。
MatMulOp
MatMulOp::Compute
OpKernelContext
我的问题是如何形成这样的输入来调用此操作?