c++ - 如何在 tensorflow 内核代码中更改变量类型张量的值？

Question

我正在尝试更改 ctx(OpKernelContext) 中的输出值在计算函数中。我可以通过 tensor->flat<>() 检查值当操作由常量执行时。

但是，当操作由变量执行时，我找不到它们。如果我这样做 flat<>() ，它返回段错误。我想在某些具有可变输入的操作中访问输出值。

Answer 1

如果您尝试了更多详细信息，这是不可能重现的。

也就是说，直接访问变量张量的内容不应导致段错误。如果您尝试在 CPU 内核中取消引用指向 GPU 内存的指针，会发生什么情况，这可能就是这里发生的情况。